Add SMT Tensor Dialect (#77)

Author: Hatsunespica

xdsl_smt/cli/xdsl_smt.py

@@ -17,6 +17,7 @@
 from xdsl_smt.dialects.effects.ub_effect import UBEffectDialect
 from xdsl_smt.dialects.effects.memory_effect import MemoryEffectDialect
 from xdsl_smt.dialects.memory_dialect import MemoryDialect
+from xdsl_smt.dialects.smt_tensor_dialect import SMTTensorDialect
 from xdsl_smt.passes.lower_effects import LowerEffectPass
 from xdsl_smt.passes.load_parametric_int_semantics import LoadIntSemanticsPass
 from xdsl_smt.passes.lower_memory_effects import LowerMemoryEffectsPass
@@ -76,6 +77,7 @@ def register_all_dialects(self):
         self.ctx.register_dialect(SMTIntDialect.name, lambda: SMTIntDialect)
         self.ctx.register_dialect(SMTBitVectorDialect.name, lambda: SMTBitVectorDialect)
         self.ctx.register_dialect(SMTArray.name, lambda: SMTArray)
+        self.ctx.register_dialect(SMTTensorDialect.name, lambda: SMTTensorDialect)
         self.ctx.register_dialect(SMTUtilsDialect.name, lambda: SMTUtilsDialect)
         self.ctx.register_dialect(EffectDialect.name, lambda: EffectDialect)
         self.ctx.register_dialect(UBEffectDialect.name, lambda: UBEffectDialect)

xdsl_smt/dialects/__init__.py

@@ -22,13 +22,15 @@ def get_smt_dialect():
         from xdsl_smt.dialects.smt_dialect import SMTDialect
         from xdsl_smt.dialects.smt_int_dialect import SMTIntDialect
         from xdsl_smt.dialects.smt_utils_dialect import SMTUtilsDialect
+        from xdsl_smt.dialects.smt_tensor_dialect import SMTTensorDialect
 
         dialects = [
             SMTArray,
             SMTBitVectorDialect,
             SMTDialect,
             SMTIntDialect,
             SMTUtilsDialect,
+            SMTTensorDialect,
         ]
         return Dialect(
             "smt",