code: def nvcc_args(self):
      args = super().nvcc_args()
      try:
          import torch
      except ImportError:
          self.warning(
              "Please install torch if trying to pre-compile kernels"
          )
          return args
      major = torch.cuda.get_device_properties(0).major  # ignore-cuda
      minor = torch.cuda.get_device_properties(0).minor  # ignore-cuda
      args.append(f"-DGPU_ARCH={major}{minor}")
      return args
  
file_path: deepspeed/op_builder/evoformer_attn.py
imports: from .builder import CUDAOpBuilder, installed_cuda_version
  import os
label_desc: Use image node_id 5 for calling the EvoformerAttnBuilder obj's underlying member method code with example usage: obj.nvcc_args() and returns: args, args