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