register TT dialect

tenstorrent · Dec 20, 2024 · 2b2d145 · 2b2d145
1 parent 9cdbd46
commit 2b2d145
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/Target/TTKernel/TTKernelToCppRegistration.cpp b/lib/Target/TTKernel/TTKernelToCppRegistration.cpp
@@ -6,6 +6,7 @@
 #include "mlir/Dialect/Func/IR/FuncOps.h"
 #include "mlir/Tools/mlir-translate/Translation.h"
 
+#include "ttmlir/Dialect/TT/IR/TT.h"
 #include "ttmlir/Dialect/TTKernel/IR/TTKernel.h"
 #include "ttmlir/Target/TTKernel/TTKernelToCpp.h"
 #include "mlir/Dialect/SCF/IR/SCF.h"
@@ -23,7 +24,7 @@ void registerTTKernelToCpp() {
       [](DialectRegistry &registry) {
         registry.insert<mlir::scf::SCFDialect,
                         mlir::tt::ttkernel::TTKernelDialect, mlir::arith::ArithDialect,
-                        mlir::emitc::EmitCDialect, mlir::func::FuncDialect>();
+                        mlir::emitc::EmitCDialect, mlir::func::FuncDialect, mlir::tt::TTDialect>();
       });
 }