unit test small change

tenstorrent · Feb 2, 2025 · 449a235 · 449a235
1 parent fc03a09
commit 449a235
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/ttmlir/Conversion/TTKernelToEmitC/ttkernel.mlir b/test/ttmlir/Conversion/TTKernelToEmitC/ttkernel.mlir
@@ -22,8 +22,10 @@ module attributes {} {
     %4 = "ttkernel.get_noc_addr_xy"(%c0_i32, %c0_i32, %c262208_i32) : (i32, i32, i32) -> !ttkernel.noc_addr
     // CHECK: emitc.call_opaque "noc_async_read"[[C:.*]]
     "ttkernel.noc_async_read"(%4, %c262432_i32, %c32_i32) : (!ttkernel.noc_addr, i32, i32) -> ()
-    // CHECK: [[C:.*]] = emitc.call_opaque "get_noc_addr_from_bank_id"[[C:.*]] {template_args = [#emitc.opaque<"true">]} [[C:.*]]
-    %5 = "ttkernel.get_noc_addr_from_bank_id"(%c32_i32, %c262432_i32) : (i32, i32) -> !ttkernel.noc_addr
+    %bank_id = arith.constant 1 : i32
+    %addr_offset = arith.constant 262400 : i32
+    %noc_addr = "ttkernel.get_noc_addr_from_bank_id"(%bank_id, %addr_offset) : (i32, i32) -> !ttkernel.noc_addr
+    // CHECK: = emitc.call_opaque "get_noc_addr_from_bank_id"({{.*}}) {template_args = [#emitc.opaque<"true">]}
     // CHECK: emitc.call_opaque "noc_async_read_barrier"[[C:.*]]
     "ttkernel.noc_async_read_barrier"() : () -> ()
     "ttkernel.return"() : () -> ()