code: def test_lower_te():
      a = te.placeholder((32, 2, 2))
      b = te.compute((32, 2, 2), lambda i, j, k: a[i, j, k] * 2.0)
      s = te.create_schedule(b.op)
      s[b].bind(b.op.axis[1], te.thread_axis("threadIdx.x"))
      s[b].bind(b.op.axis[2], te.thread_axis("threadIdx.x"))
      orig_mod = tvm.driver.build_module.schedule_to_module(s, [a, b])
      mod = tvm.tir.transform.UnifyThreadBinding()(orig_mod)
      tvm.ir.assert_structural_equal(mod, orig_mod)
  
imports: import pytest
  import sys
  import tvm
  import tvm.testing
  from tvm import te
  from tvm.script import tir
label_desc: Use image node_id 24 for calling a global function with example usage: test_lower_te() without return types