repro #1942

llvm-lit -sv test/ttmlir/Silicon/TTNN/repro_mean.mlir ttrt run build/test/ttmlir/Silicon/TTNN/Output/repro_mean.mlir.tmp.ttnn
tenstorrent · Jan 23, 2025 · 8a0ab1b · 8a0ab1b
1 parent 9e5980e
commit 8a0ab1b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/ttmlir/Silicon/TTNN/repro_mean.mlir b/test/ttmlir/Silicon/TTNN/repro_mean.mlir
@@ -0,0 +1,8 @@
+// RUN: ttmlir-opt --ttir-to-ttnn-backend-pipeline="system-desc-path=%system_desc_path%" %s > %t.mlir
+// RUN: ttmlir-translate --ttnn-to-flatbuffer %t.mlir > %t.ttnn
+
+func.func @forward(%arg0: tensor<512x1024xbf16>) -> tensor<512x32xbf16> {
+  %0 = tensor.empty() : tensor<512x32xbf16>
+  %1 = "ttir.mean"(%arg0, %0) <{dim_arg = [-1: i32], keep_dim = true}> : (tensor<512x1024xbf16>, tensor<512x32xbf16>) -> tensor<512x32xbf16>
+  return %1 : tensor<512x32xbf16>
+}