code: def check_batch_matmul(call):
      """Check if the given batch_matmul workload can be offloaded to CUTLASS."""
      batch_matmul = get_root_call(call, "nn.batch_matmul")
      lhs = batch_matmul.args[0].checked_type
      rhs = batch_matmul.args[1].checked_type
      transpose_a = batch_matmul.attrs.transpose_a
      transpose_b = batch_matmul.attrs.transpose_b
      return check_dtype(lhs, rhs) and not transpose_a and transpose_b
  
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 11 for calling a global function with example usage: check_batch_matmul(call) and returns: unknown