code: def append_nvcc_threads(nvcc_extra_args):
      _, bare_metal_version = get_cuda_bare_metal_version(CUDA_HOME)
      if bare_metal_version >= Version("11.2"):
          return nvcc_extra_args + ["--threads", "4"]
      return nvcc_extra_args
  
file_path: flash-attention/csrc/rotary/setup.py
imports: import sys
  import warnings
  import os
  from packaging.version import parse, Version
  import torch
  from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension, CUDA_HOME
  from setuptools import setup, find_packages
  import subprocess
label_desc: Use image node_id 4 for calling a global function with example usage: append_nvcc_threads(nvcc_extra_args) and returns: nvcc_extra_args, unknown