code: def _check_workload(
      te_workload, tir_workload, index_dtype_override=None
  ):
      func = te.create_prim_func(te_workload(), index_dtype_override)
      tvm.ir.assert_structural_equal(func, tir_workload)
      # make sure that we can create schedule from the func
      s = tir.Schedule(func, debug_mask="all")
      assert s
  
imports: import numpy
  import tvm
  import tvm.testing
  from tvm import te, tir, topi, relay
  from tvm.script import tir
  import pytest
label_desc: Use image node_id 3 for calling a global function with example usage: _check_workload(te_workload, tir_workload, index_dtype_override) without return types