code: def _replace_stmt_with_buf_var_names(buffer_info_map):
      """helper to replace tir.allocates with buffer names"""
      new_buffer_info_map = dict()
      for k, v in buffer_info_map.items():
          new_buffer_info_map[v.buffer_var.name] = k
      return new_buffer_info_map
  
imports: import pytest
  import tvm
  from tvm import tir, script
  from tvm.script import tir
  from tvm.tir import stmt_functor
  from tvm.tir.usmp import utils
  from tvm.target import Target
  from tvm import WorkspacePoolInfo, PoolInfoProperties
label_desc: Use image node_id 1 for calling a global function with example usage: _replace_stmt_with_buf_var_names(buffer_info_map) and returns: new_buffer_info_map