We don't support complex types in tcp.mul today. Without this, it fai…

…ls the TCP e2e integration test for ElementwiseMulTensorComplexModule_basic...
llvm · Sep 15, 2023 · 4d90ab4 · 4d90ab4
1 parent 158ec7d
commit 4d90ab4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...xternal-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/Tcp/IR/TcpOps.td b/...xternal-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)";