Skip to content

Commit

Permalink
interpreter: sort interpreter functions registering (#3688)
Browse files Browse the repository at this point in the history
A tiny bit of cleanup.
  • Loading branch information
superlopuh authored Jan 2, 2025
1 parent 91eeb53 commit 19abdc8
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions xdsl/interpreters/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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())

0 comments on commit 19abdc8

Please sign in to comment.