code: def test_mma_auto_tensorization():
      mod = te.create_prim_func(
          matmul_fp16(M=4096, N=4096, K=4096, out_dtype="float16")
      )
      actual = _design_space(mod, "float16")
      check_sketches(
          mod,
          sketches=actual,
          expected_mods=[MmaModule],
          expected_decisions=[gemm_decision],
      )
  
imports: import tempfile
  import numpy
  import tvm
  from tvm import te
  from tvm import meta_schedule
  from tvm._ffi import register_func
  from tvm.meta_schedule.testing.space_generation import check_sketches, generate_design_space
  from tvm.meta_schedule.builder import LocalBuilder
  from tvm.script import ir
  from tvm.script import tir
  from tvm.target import Target
  from tvm.tir import Schedule
  from tvm.tir.schedule import Trace
  from tvm.tir.tensor_intrin import cuda
  import tvm.testing
label_desc: Use image node_id 4 for calling a global function with example usage: test_mma_auto_tensorization() without return types