code: def append_nvcc_threads(nvcc_extra_args):
      (
          _,
          bare_metal_major,
          bare_metal_minor,
      ) = get_cuda_bare_metal_version(CUDA_HOME)
      if int(bare_metal_major) >= 11 and int(bare_metal_minor) >= 2:
          return nvcc_extra_args + ["--threads", "4"]
      return nvcc_extra_args
  
imports: import os
  import subprocess
  import torch
  from setuptools import setup
  from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CUDA_HOME
label_desc: Use image node_id 2 for calling a global function with example usage: append_nvcc_threads(nvcc_extra_args) and returns: nvcc_extra_args, unknown