From 602fa4e983c271af1178d179d97ab5156f7e0fe7 Mon Sep 17 00:00:00 2001
From: Stefan Djordjevic <sdjordjevic@tenstorrent.com>
Date: Thu, 26 Dec 2024 11:25:44 +0000
Subject: [PATCH] Adding a repro test to verify revert

---
 test/ttmlir/Silicon/TTNN/matmul/llama_matmul.mlir     | 11 +++++++++++
 .../Silicon/TTNN/{ => matmul}/simple_matmul.mlir      |  0
 2 files changed, 11 insertions(+)
 create mode 100644 test/ttmlir/Silicon/TTNN/matmul/llama_matmul.mlir
 rename test/ttmlir/Silicon/TTNN/{ => matmul}/simple_matmul.mlir (100%)

diff --git a/test/ttmlir/Silicon/TTNN/matmul/llama_matmul.mlir b/test/ttmlir/Silicon/TTNN/matmul/llama_matmul.mlir
new file mode 100644
index 000000000..e777cd55a
--- /dev/null
+++ b/test/ttmlir/Silicon/TTNN/matmul/llama_matmul.mlir
@@ -0,0 +1,11 @@
+// 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
+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 b/test/ttmlir/Silicon/TTNN/matmul/simple_matmul.mlir
similarity index 100%
rename from test/ttmlir/Silicon/TTNN/simple_matmul.mlir
rename to test/ttmlir/Silicon/TTNN/matmul/simple_matmul.mlir