code: def test_tir_builtin_expression():
      dims = (128, 128)
  
      @T.prim_func(private=True)
      def with_builtin(a: T.handle) -> None:
          A = T.match_buffer(a, [len(dims), *dims], "int32")
          for i, j, k in T.grid(*A.shape):
              A[i, j, k] = T.int32(1 + len(A.shape))
  
      @T.prim_func(private=True)
      def evaluated(A: T.Buffer((2, 128, 128), "int32")):
          for i, j, k in T.grid(2, 128, 128):
              A[i, j, k] = 4
  
      tvm.ir.assert_structural_equal(with_builtin, evaluated)
  
imports: import pytest
  import tvm.testing
  from tvm.script.parser import tir
  from tvm import ir, tir
label_desc: Use image node_id 15 for calling a global function with example usage: test_tir_builtin_expression() without return types