try adding arith

tenstorrent · Nov 20, 2024 · 1b8479d · 1b8479d
1 parent 0c7aeb0
commit 1b8479d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/lib/Conversion/TTIRToLinAlg/CMakeLists.txt b/lib/Conversion/TTIRToLinAlg/CMakeLists.txt
@@ -12,4 +12,5 @@ add_mlir_library(TTMLIRTTIRToLinAlg
   MLIRIR
   MLIRPass
   MLIRLinalgDialect
+  MLIRArithmeticDialect
 )
diff --git a/lib/Conversion/TTIRToLinAlg/TTIRToLinAlgPass.cpp b/lib/Conversion/TTIRToLinAlg/TTIRToLinAlgPass.cpp
@@ -4,6 +4,7 @@
 
 #include "ttmlir/Conversion/TTIRToLinAlg/TTIRToLinAlg.h"
 
+#include "mlir/Dialect/Arithmetic/IR/Arithmetic.h"
 #include "mlir/Dialect/Func/Transforms/FuncConversions.h"
 #include "mlir/IR/BuiltinDialect.h"
 #include "mlir/IR/PatternMatch.h"
@@ -12,6 +13,7 @@
 #include "mlir/Transforms/DialectConversion.h"
 #include "ttmlir/Conversion/TTIRToTTIRDecomposition/TTIRToTTIRDecomposition.h"
 #include "ttmlir/Dialect/TTIR/IR/TTIR.h"
+
 #include <mlir/Dialect/Func/IR/FuncOps.h>
 
 using namespace mlir;
@@ -34,6 +36,7 @@ struct ConvertTTIRToLinAlgPass
     target.addLegalDialect<func::FuncDialect>();
     target.addLegalDialect<tensor::TensorDialect>();
     target.addLegalDialect<linalg::LinalgDialect>();
+    target.addLegalDialect<linalg::LinalgDialect>();
     target.addIllegalDialect<ttir::TTIRDialect>();
 
     TypeConverter typeConverter;