Adding a repro test to verify revert

tenstorrent · Dec 26, 2024 · 40d79fe · 40d79fe
1 parent df24852
commit 40d79fe
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/test/ttmlir/Silicon/TTNN/matmul/llama_matmul.mlir b/test/ttmlir/Silicon/TTNN/matmul/llama_matmul.mlir
@@ -0,0 +1,12 @@
+// RUN: ttmlir-opt --ttir-to-ttnn-backend-pipeline="system-desc-path=%system_desc_path%" %s > %t.mlir
+// RUN: FileCheck %s --input-file=%t.mlir
+// RUN: ttmlir-translate --ttnn-to-flatbuffer %t.mlir > %t.ttnn
+// CHECK: #[[TILED_LAYOUT:.*]] = #ttnn.ttnn_layout<(d0, d1) -> (d0, d1), <1x1>, memref<2x4x!tt.tile<32x32, bf16>, #dram>, <interleaved>>
+module attributes {} {
+  func.func @forward(%arg0: tensor<1x11x2048xf32>, %arg1: tensor<2048x128256xf32>) -> tensor<1x11x128256xf32> {
+    %0 = tensor.empty() : tensor<1x11x128256xf32>
+    // CHECK: %[[C:.*]] = "ttnn.matmul"[[C:.*]]
+    %1 = "ttir.matmul"(%arg0, %arg1, %0) : (tensor<1x11x2048xf32>, tensor<2048x128256xf32>, tensor<1x11x128256xf32>) -> tensor<1x11x128256xf32>
+    return %1 : tensor<1x11x128256xf32>
+  }
+}
diff --git a/test/ttmlir/Silicon/TTNN/simple_matmul.mlir → ...ir/Silicon/TTNN/matmul/simple_matmul.mlir b/test/ttmlir/Silicon/TTNN/simple_matmul.mlir → ...ir/Silicon/TTNN/matmul/simple_matmul.mlir