code: def test_specialize_decl_buffer():
      """Buffers occurring in a DeclBuffer statement should be updated"""
  
      @T.prim_func(private=True)
      def before(A_data: T.handle("float32"), A_size: T.int32):
          A_buf = T.decl_buffer(A_size, "float32", data=A_data)
          for i in range(A_size):
              A_buf[i] = A_buf[i] * 2.0
  
      @T.prim_func(private=True)
      def expected(A_data: T.handle("float32")):
          A_buf = T.decl_buffer(16, "float32", data=A_data)
          for i in range(16):
              A_buf[i] = A_buf[i] * 2.0
  
      param_map = {before.params[1]: T.int32(16)}
      after = before.specialize(param_map)
  
      tvm.ir.assert_structural_equal(expected, after)
  
imports: import tvm
  from tvm.script import tir
  from tvm.tir.schedule.testing import assert_structural_equal_ignore_global_symbol
label_desc: Use image node_id 17 for calling a global function with example usage: test_specialize_decl_buffer() without return types