Marked func::CallOp as legal in StableHLOToTTIRPass and in ArithToSta…

…bleHLOPass
tenstorrent · Nov 5, 2024 · d425472 · d425472
1 parent 3dbf089
commit d425472
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp b/lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp
@@ -59,6 +59,7 @@ struct ConvertArithToStableHLOPass
     target.addLegalOp<mlir::ModuleOp>();
     target.addLegalOp<mlir::func::FuncOp>();
     target.addLegalOp<mlir::func::ReturnOp>();
+    target.addLegalOp<mlir::func::CallOp>();
 
     // For now keep the same type assuming StableHLO ops operate on builtin
     // tensor.

diff --git a/lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPass.cpp b/lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPass.cpp
@@ -79,6 +79,7 @@ struct ConvertStableHLOToTTIRPass
     target.addLegalOp<mlir::ModuleOp>();
     target.addLegalOp<mlir::func::FuncOp>();
     target.addLegalOp<mlir::func::ReturnOp>();
+    target.addLegalOp<mlir::func::CallOp>();
 
     // For now keep the same type assuming StableHLO ops operate on builtin
     // tensor.