code: def just_cutlass(model):
      """Run partition_for_cutlass, complete the compilation with TVM, and profile the result."""
      logging.info(f"just_cutlass | {model['name']}")
      logging.info("-------------- BEGIN ORIGINAL --------------")
      logging.info(model["mod"])
      logging.info("-------------- END ORIGINAL ----------------")
      tmp_dir = tempfile.mkdtemp()
      autotvm_tune_module(model["mod"], CUDA, TUNING_LOG)
      with optional_tuning_records(TUNING_LOG):
          with tvm.transform.PassContext(
              opt_level=3, disabled_pass=["AlterOpLayout"]
          ):
              logging.info("Partitioning for CUTLASS...")
              mod = tvm.relay.op.contrib.partition_for_cutlass(
                  model["mod"], model["params"]
              )
              partitioned_model = model.copy()
              partitioned_model["mod"] = mod
              logging.info(
                  "-------------- BEGIN PARTITIONED --------------"
              )
              logging.info(partitioned_model["mod"])
              logging.info(
                  "-------------- END PARTITIONED ----------------"
              )
              targets = []
              targets.append(CUDA)
              targets.append(
                  tvm.target.Target(f"cutlass -tmp_dir={tmp_dir}", HOST)
              )
              dev = tvm.device(CUDA.get_target_device_type())
              compile_and_benchmark(
                  "just_cutlass",
                  partitioned_model,
                  targets,
                  dev,
                  tmp_dir,
              )
  
imports: import tvm
  import logging
  import tempfile
  import os
  import shutil
  import menangerie
  from tvm.relay.op.contrib.cutlass import partition_for_cutlass
  from tvm.contrib.cutlass import num_cutlass_partitions
  from tvm.relay.op.contrib.cublas import partition_for_cublas
  from tvm.relay.op.contrib.cudnn import partition_for_cudnn
label_desc: Use image node_id 9 for calling a global function with example usage: just_cutlass(model) without return types