code: def extract_layout_transform_task(
      mod: tvm.IRModule, target: tvm.target.Target
  ) -> meta_schedule.ExtractedTask:
      """Given a relay IRModule, return the PrimFunc IRModule with fused layout transform task."""
      extracted_tasks = meta_schedule.relay_integration.extract_tasks(
          mod,
          target,
          {},
          pass_config={"relay.backend.use_meta_schedule": True},
      )
      task_of_interest = None
      for task in extracted_tasks:
          if "layout_transform" in task.task_name:
              task_of_interest = task
              break
      assert task_of_interest is not None
      return task_of_interest
  
imports: import itertools
  import random
  import tempfile
  from typing import Callable, Dict, List, Optional, Tuple, Union
  import numpy
  import pytest
  import tvm
  import tvm.testing
  from tvm import meta_schedule, relay
  from tvm.meta_schedule.schedule.cuda.layout_transform import cuda_layout_transform_schedule_rule
  from tvm.relay.op import OpPattern
  from tvm.script import ir
  from tvm.script import tir
  from tvm.tir.schedule import BlockRV
label_desc: Use image node_id 8 for calling a global function with example usage: extract_layout_transform_task(mod, target) and returns: task_of_interest