code: def load_test_module(name):
      import sys
      from importlib.machinery import SourceFileLoader
      from pathlib import Path
      from unittest import mock
  
      testdir = Path(__file__).absolute().parent.parent
      with mock.patch("sys.path", [*sys.path, str(testdir)]):
          return SourceFileLoader(
              name, str(testdir / f"{name.replace('.', '/')}.py")
          ).load_module()
  
imports: import unittest
  from typing import List
  import torch
  import torch.distributed
  from torch._C import FileCheck
  from torch._dynamo.utils import same
  from torch._inductor.utils import fresh_inductor_cache, run_and_get_triton_code
  from torch.testing._internal.common_distributed import MultiProcessTestCase, requires_nccl, skip_if_lt_x_gpu
  from torch.testing._internal.common_utils import run_tests
  from torch.utils._triton import has_triton
  import sys
label_desc: Use image node_id 1 for calling a global function with example usage: load_test_module(name) and returns: SourceFileLoader