From d4254720ae24be753dc0281355030aa401cba64a Mon Sep 17 00:00:00 2001
From: Kristijan Mitrovic <kmitrovic@tenstorrent.com>
Date: Mon, 4 Nov 2024 16:42:11 +0000
Subject: [PATCH] Marked func::CallOp as legal in StableHLOToTTIRPass and in
 ArithToStableHLOPass

---
 lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp | 1 +
 lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPass.cpp  | 1 +
 2 files changed, 2 insertions(+)

diff --git a/lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp b/lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp
index 587847582..e759170e8 100644
--- 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
index 5a910dd61..b6c0b4988 100644
--- 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.