From 8d2f32a9f60a0649b970bc8c49b9f1e06455008b Mon Sep 17 00:00:00 2001
From: uazizTT <uaziz@tenstorrent.com>
Date: Mon, 21 Oct 2024 11:41:50 -0400
Subject: [PATCH] Legalize ttir EmptyOp.

---
 lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp b/lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp
index 4e058825c6..5878475822 100644
--- a/lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp
+++ b/lib/Conversion/StableHLOToTTIR/ArithToStableHLOPass.cpp
@@ -55,6 +55,7 @@ struct ConvertArithToStableHLOPass
 
     target.addIllegalDialect<mlir::arith::ArithDialect>();
     target.addLegalDialect<mlir::stablehlo::StablehloDialect>();
+    target.addLegalOp<mlir::tensor::EmptyOp>();
     target.addLegalOp<mlir::ModuleOp>();
     target.addLegalOp<mlir::func::FuncOp>();
     target.addLegalOp<mlir::func::ReturnOp>();