code: def write_and_load_module(code, num_extra_lines):
      with tempfile.NamedTemporaryFile(mode="w+", suffix=".py") as f:
          f.write(("# extra line\n" * num_extra_lines) + code)
          f.flush()
          spec = importlib.util.spec_from_file_location(
              "module.name", f.name
          )
          module = importlib.util.module_from_spec(spec)
          spec.loader.exec_module(module)
      return module
  
imports: import importlib.util
  import os
  import shutil
  import tempfile
  import pytest
  import torch
  import triton
  import triton.language
  from triton.runtime.jit import JITFunction
label_desc: Use image node_id 9 for calling a global function with example usage: write_and_load_module(code, num_extra_lines) and returns: module