code: def test_replace_block_remap():
      func = elementwise
      s = tir.ScheduleState(func, debug_mask="all")
      # The target stmt
      target = matmul.body.block.body.body.body[0].block
      sref = s.get_sref(
          s.mod["main"].body.block.body[0].body.body.block
      )
      s.replace(sref, target, {sref.stmt: target})
      sref_new = s.get_sref(
          s.mod["main"].body.block.body[0].body.body.block
      )
      # Check the original sref has been remapped
      assert sref.__hash__() == sref_new.__hash__()
      tvm.ir.assert_structural_equal(sref.stmt, target)
  
imports: import gc
  import sys
  import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.ir import IRModule
  from tvm.script import tir
label_desc: Use image node_id 17 for calling a global function with example usage: test_replace_block_remap() without return types