def get_all_dialects() -> dict[str, Callable[[], Dialect]]:
"""Returns all available dialects."""
def get_acc():
from xdsl.dialects.acc import ACC
return ACC
def get_accfg():
from xdsl.dialects.accfg import ACCFG
return ACCFG
def get_affine():
from xdsl.dialects.affine import Affine
return Affine
def get_air():
from xdsl.dialects.experimental.air import AIR
return AIR
def get_arith():
from xdsl.dialects.arith import Arith
return Arith
def get_arm():
from xdsl.dialects.arm import ARM
return ARM
def get_arm_func():
from xdsl.dialects.arm_func import ARM_FUNC
return ARM_FUNC
def get_arm_neon():
from xdsl.dialects.arm_neon import ARM_NEON
return ARM_NEON
def get_asm():
from xdsl.dialects.asm import ASM
return ASM
def get_bigint():
from xdsl.dialects.bigint import BigInt
return BigInt
def get_bufferization():
from xdsl.dialects.bufferization import Bufferization
return Bufferization
def get_builtin():
from xdsl.dialects.builtin import Builtin
return Builtin
def get_cf():
from xdsl.dialects.cf import Cf
return Cf
def get_cmath():
from xdsl.dialects.cmath import Cmath
return Cmath
def get_comb():
from xdsl.dialects.comb import Comb
return Comb
def get_complex():
from xdsl.dialects.complex import Complex
return Complex
def get_csl():
from xdsl.dialects.csl import CSL
return CSL
def get_csl_stencil():
from xdsl.dialects.csl_stencil import CSL_STENCIL
return CSL_STENCIL
def get_csl_wrapper():
from xdsl.dialects.csl_wrapper import CSL_WRAPPER
return CSL_WRAPPER
def get_dlti():
from xdsl.dialects.dlti import DLTI
return DLTI
def get_dmp():
from xdsl.dialects.experimental.dmp import DMP
return DMP
def get_ematch():
from xdsl.dialects.ematch import Ematch
return Ematch
def get_emitc():
from xdsl.dialects.emitc import EmitC
return EmitC
def get_equivalence():
from xdsl.dialects.equivalence import Equivalence
return Equivalence
def get_eqsat_pdl_interp():
from xdsl.dialects.eqsat_pdl_interp import EqSatPDLInterp
return EqSatPDLInterp
def get_fir():
from xdsl.dialects.experimental.fir import FIR
return FIR
def get_fsm():
from xdsl.dialects.fsm import FSM
return FSM
def get_func():
from xdsl.dialects.func import Func
return Func
def get_gpu():
from xdsl.dialects.gpu import GPU
return GPU
def get_hlfir():
from xdsl.dialects.experimental.hlfir import HLFIR
return HLFIR
def get_hls():
from xdsl.dialects.experimental.hls import HLS
return HLS
def get_hw():
from xdsl.dialects.hw import HW
return HW
def get_linalg():
from xdsl.dialects.linalg import Linalg
return Linalg
def get_irdl():
from xdsl.dialects.irdl.irdl import IRDL
return IRDL
def get_llvm():
from xdsl.dialects.llvm import LLVM
return LLVM
def get_ltl():
from xdsl.dialects.ltl import LTL
return LTL
def get_math():
from xdsl.dialects.math import Math
return Math
def get_math_xdsl():
from xdsl.dialects.math_xdsl import MathXDSL
return MathXDSL
def get_memref():
from xdsl.dialects.memref import MemRef
return MemRef
def get_memref_stream():
from xdsl.dialects.memref_stream import MemRefStream
return MemRefStream
def get_ml_program():
from xdsl.dialects.ml_program import MLProgram
return MLProgram
def get_mod_arith():
from xdsl.dialects.mod_arith import ModArith
return ModArith
def get_mpi():
from xdsl.dialects.mpi import MPI
return MPI
def get_omp():
from xdsl.dialects.omp import OMP
return OMP
def get_pdl():
from xdsl.dialects.pdl import PDL
return PDL
def get_pdl_interp():
from xdsl.dialects.pdl_interp import PDLInterp
return PDLInterp
def get_polynomial():
from xdsl.dialects.polynomial import Polynomial
return Polynomial
def get_printf():
from xdsl.dialects.printf import Printf
return Printf
def get_ptr_xdsl():
from xdsl.dialects.ptr import Ptr
return Ptr
def get_py():
from xdsl.dialects.py import Py
return Py
def get_riscv_debug():
from xdsl.dialects.riscv_debug import RISCV_Debug
return RISCV_Debug
def get_riscv():
from xdsl.dialects.riscv import RISCV
return RISCV
def get_RV32():
from xdsl.dialects.rv32 import RV32
return RV32
def get_RV64():
from xdsl.dialects.rv64 import RV64
return RV64
def get_riscv_func():
from xdsl.dialects.riscv_func import RISCV_Func
return RISCV_Func
def get_riscv_scf():
from xdsl.dialects.riscv_scf import RISCV_Scf
return RISCV_Scf
def get_riscv_cf():
from xdsl.dialects.riscv_cf import RISCV_Cf
return RISCV_Cf
def get_riscv_snitch():
from xdsl.dialects.riscv_snitch import RISCV_Snitch
return RISCV_Snitch
def get_scf():
from xdsl.dialects.scf import Scf
return Scf
def get_seq():
from xdsl.dialects.seq import Seq
return Seq
def get_shard():
from xdsl.dialects.shard import Shard
return Shard
def get_smt():
from xdsl.dialects.smt import SMT
return SMT
def get_snitch():
from xdsl.dialects.snitch import Snitch
return Snitch
def get_snitch_runtime():
from xdsl.dialects.snitch_runtime import SnitchRuntime
return SnitchRuntime
def get_snitch_stream():
from xdsl.dialects.snitch_stream import SnitchStream
return SnitchStream
def get_stencil():
from xdsl.dialects.stencil import Stencil
return Stencil
def get_symref():
from xdsl.dialects.symref import Symref
return Symref
def get_tensor():
from xdsl.dialects.tensor import Tensor
return Tensor
def get_test():
from xdsl.dialects.test import Test
return Test
def get_tosa():
from xdsl.dialects.tosa import TOSA
return TOSA
def get_transform():
from xdsl.dialects.transform import Transform
return Transform
def get_ub():
from xdsl.dialects.ub import UB
return UB
def get_varith():
from xdsl.dialects.varith import Varith
return Varith
def get_vector():
from xdsl.dialects.vector import Vector
return Vector
def get_wasm():
from xdsl.dialects.wasm import Wasm
return Wasm
def get_wasmssa():
from xdsl.dialects.wasmssa import WasmSSA
return WasmSSA
def get_x86():
from xdsl.dialects.x86 import X86
return X86
def get_x86_func():
from xdsl.dialects.x86_func import X86_FUNC
return X86_FUNC
def get_x86_scf():
from xdsl.dialects.x86_scf import X86_Scf
return X86_Scf
return {
"acc": get_acc,
"accfg": get_accfg,
"affine": get_affine,
"air": get_air,
"arith": get_arith,
"arm": get_arm,
"arm_func": get_arm_func,
"arm_neon": get_arm_neon,
"asm": get_asm,
"bigint": get_bigint,
"bufferization": get_bufferization,
"builtin": get_builtin,
"cf": get_cf,
"cmath": get_cmath,
"comb": get_comb,
"complex": get_complex,
"csl": get_csl,
"csl_stencil": get_csl_stencil,
"csl_wrapper": get_csl_wrapper,
"dlti": get_dlti,
"dmp": get_dmp,
"ematch": get_ematch,
"emitc": get_emitc,
"equivalence": get_equivalence,
"eqsat_pdl_interp": get_eqsat_pdl_interp,
"fir": get_fir,
"fsm": get_fsm,
"func": get_func,
"gpu": get_gpu,
"hlfir": get_hlfir,
"hls": get_hls,
"hw": get_hw,
"linalg": get_linalg,
"irdl": get_irdl,
"llvm": get_llvm,
"ltl": get_ltl,
"math": get_math,
"math_xdsl": get_math_xdsl,
"memref": get_memref,
"memref_stream": get_memref_stream,
"ml_program": get_ml_program,
"mod_arith": get_mod_arith,
"mpi": get_mpi,
"omp": get_omp,
"pdl": get_pdl,
"pdl_interp": get_pdl_interp,
"polynomial": get_polynomial,
"printf": get_printf,
"ptr_xdsl": get_ptr_xdsl,
"py": get_py,
"riscv": get_riscv,
"rv32": get_RV32,
"rv64": get_RV64,
"riscv_debug": get_riscv_debug,
"riscv_func": get_riscv_func,
"riscv_scf": get_riscv_scf,
"riscv_cf": get_riscv_cf,
"riscv_snitch": get_riscv_snitch,
"scf": get_scf,
"seq": get_seq,
"shard": get_shard,
"smt": get_smt,
"snitch": get_snitch,
"snrt": get_snitch_runtime,
"snitch_stream": get_snitch_stream,
"stencil": get_stencil,
"symref": get_symref,
"tensor": get_tensor,
"test": get_test,
"tosa": get_tosa,
"ub": get_ub,
"varith": get_varith,
"vector": get_vector,
"wasm": get_wasm,
"wasmssa": get_wasmssa,
"x86": get_x86,
"x86_func": get_x86_func,
"x86_scf": get_x86_scf,
"transform": get_transform,
}