From 19abdc8361f5cd0ff180153ccfd51cf9dfc81ec7 Mon Sep 17 00:00:00 2001 From: Sasha Lopoukhine Date: Thu, 2 Jan 2025 15:50:31 +0000 Subject: [PATCH] interpreter: sort interpreter functions registering (#3688) A tiny bit of cleanup. --- xdsl/interpreters/__init__.py | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/xdsl/interpreters/__init__.py b/xdsl/interpreters/__init__.py index db2b5f1aad..1116a3f53d 100644 --- a/xdsl/interpreters/__init__.py +++ b/xdsl/interpreters/__init__.py @@ -31,20 +31,26 @@ def register_implementations( include_wgpu: bool = True, include_onnx: bool = True, ): - interpreter.register_implementations(func.FuncFunctions()) + interpreter.register_implementations(affine.AffineFunctions()) + interpreter.register_implementations(arith.ArithFunctions()) + interpreter.register_implementations(builtin.BuiltinFunctions()) interpreter.register_implementations(cf.CfFunctions()) - interpreter.register_implementations(riscv.RiscvFunctions()) + interpreter.register_implementations(func.FuncFunctions()) + interpreter.register_implementations(linalg.LinalgFunctions()) + interpreter.register_implementations(memref_stream.MemrefStreamFunctions()) + interpreter.register_implementations(memref.MemrefFunctions()) + interpreter.register_implementations(ml_program.MLProgramFunctions()) + interpreter.register_implementations(pdl.PDLRewriteFunctions(ctx)) + interpreter.register_implementations(printf.PrintfFunctions()) interpreter.register_implementations(riscv_debug.RiscvDebugFunctions()) interpreter.register_implementations(riscv_func.RiscvFuncFunctions()) interpreter.register_implementations(riscv_libc.RiscvLibcFunctions()) interpreter.register_implementations(riscv_scf.RiscvScfFunctions()) interpreter.register_implementations(riscv_snitch.RiscvSnitchFunctions()) - interpreter.register_implementations(pdl.PDLRewriteFunctions(ctx)) - interpreter.register_implementations(affine.AffineFunctions()) - interpreter.register_implementations(linalg.LinalgFunctions()) - interpreter.register_implementations(memref.MemrefFunctions()) - interpreter.register_implementations(memref_stream.MemrefStreamFunctions()) - interpreter.register_implementations(ml_program.MLProgramFunctions()) + interpreter.register_implementations(riscv.RiscvFunctions()) + interpreter.register_implementations(scf.ScfFunctions()) + interpreter.register_implementations(snitch_stream.SnitchStreamFunctions()) + interpreter.register_implementations(tensor.TensorFunctions()) if include_onnx: from xdsl.interpreters import onnx @@ -54,9 +60,3 @@ def register_implementations( from xdsl.interpreters import wgpu interpreter.register_implementations(wgpu.WGPUFunctions()) - interpreter.register_implementations(builtin.BuiltinFunctions()) - interpreter.register_implementations(arith.ArithFunctions()) - interpreter.register_implementations(printf.PrintfFunctions()) - interpreter.register_implementations(scf.ScfFunctions()) - interpreter.register_implementations(snitch_stream.SnitchStreamFunctions()) - interpreter.register_implementations(tensor.TensorFunctions())