From 4d90ab4e0f1d499777474bd8d19c5e3d60b1c82b Mon Sep 17 00:00:00 2001
From: Sambhav Jain <sambhav.jain@getcruise.com>
Date: Fri, 15 Sep 2023 10:10:53 -0700
Subject: [PATCH] We don't support complex types in tcp.mul today. Without
 this, it fails the TCP e2e integration test for
 ElementwiseMulTensorComplexModule_basic...

---
 .../include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td    | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td b/externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td
index 409febe70cb..ed26382d11d 100644
--- a/externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td
+++ b/externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td
@@ -129,12 +129,12 @@ def Tcp_MulOp : Tcp_BinaryElementwiseOp<"mul", [SameOperandsAndResultElementType
   }];
 
   let arguments = (ins
-    Tcp_Tensor:$in1,
-    Tcp_Tensor:$in2
+    Tcp_FloatOrIntTensor:$in1,
+    Tcp_FloatOrIntTensor:$in2
   );
 
   let results = (outs
-    Tcp_Tensor:$out
+    Tcp_FloatOrIntTensor:$out
   );
 
   let assemblyFormat = "$in1 `,` $in2 attr-dict `:` type($in1) `,` type($in2) `->` type($out)";