code: def check_gemm(call):
      """Check if the given dense workload can be offloaded to CUTLASS."""
      dense = get_root_call(call, "nn.dense")
      lhs = dense.args[0].checked_type
      rhs = dense.args[1].checked_type
      return check_dtype(lhs, rhs)
  
imports: from functools import partial
  from tvm import relay
  from tvm.ir.transform import PassContext, Sequential
  from tvm.relay import transform
  from tvm.relay.build_module import bind_params_by_name
  from tvm.relay.op.contrib.register import register_pattern_table
  from ...dataflow_pattern import is_constant, is_op, wildcard
label_desc: Use image node_id 10 for calling a global function with example usage: check_gemm(call) and returns: check_dtype