unit test for get_noc_addr_from_back_id to emitc

tenstorrent · Feb 2, 2025 · fc03a09 · fc03a09
1 parent ac41793
commit fc03a09
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/ttmlir/Conversion/TTKernelToEmitC/ttkernel.mlir b/test/ttmlir/Conversion/TTKernelToEmitC/ttkernel.mlir
@@ -22,6 +22,8 @@ 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
     // CHECK: emitc.call_opaque "noc_async_read_barrier"[[C:.*]]
     "ttkernel.noc_async_read_barrier"() : () -> ()
     "ttkernel.return"() : () -> ()