class_docstr: Tests just-in-time cpp extensions.
  Don't confuse this with the PyTorch JIT (aka TorchScript).
code: def test_jit_cuda_extension(self):
      # NOTE: The name of the extension must equal the name of the module.
      module = torch.utils.cpp_extension.load(
          name="torch_test_cuda_extension",
          sources=[
              "cpp_extensions/cuda_extension.cpp",
              "cpp_extensions/cuda_extension.cu",
          ],
          extra_cuda_cflags=["-O2"],
          verbose=True,
          keep_intermediates=False,
      )
  
      x = torch.zeros(100, device="cuda", dtype=torch.float32)
      y = torch.zeros(100, device="cuda", dtype=torch.float32)
  
      z = module.sigmoid_add(x, y).cpu()
  
      # 2 * sigmoid(0) = 2 * 0.5 = 1
      self.assertEqual(z, torch.ones_like(z))
  
imports: import os
  import shutil
  import sys
  import unittest
  import warnings
  import re
  import tempfile
  import subprocess
  import glob
  import torch.testing._internal.common_utils
  from torch.testing._internal.common_cuda import TEST_CUDNN, TEST_CUDA
  import torch
  import torch.backends.cudnn
  import torch.utils.cpp_extension
  from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME
  from torch.testing._internal.common_utils import gradcheck
  import torch.multiprocessing
  from torch.utils.cpp_extension import _TORCH_PATH, remove_extension_h_precompiler_headers, get_cxx_compiler, check_compiler_is_gcc
label_desc: Use image node_id 6 for calling the TestCppExtensionJIT obj's underlying member method code with example usage: obj.test_jit_cuda_extension() without return types