code: def get_cuda_cc_flag() -> List[str]:
      """
      This function produces the cc flags for your GPU arch
  
      Returns:
          The CUDA cc flags for compilation.
      """
  
      # only import torch when needed
      # this is to avoid importing torch when building on a machine without torch pre-installed
      # one case is to build wheel for pypi release
      import torch
  
      cc_flag = []
      max_arch = "".join(
          str(i) for i in torch.cuda.get_device_capability()
      )
      for arch in torch.cuda.get_arch_list():
          res = re.search(r"sm_(\d+)", arch)
          if res:
              arch_cap = res[1]
              if int(arch_cap) >= 60 and int(arch_cap) <= int(max_arch):
                  cc_flag.extend(
                      [
                          "-gencode",
                          f"arch=compute_{arch_cap},code={arch}",
                      ]
                  )
      return cc_flag
  
file_path: ColossalAI/colossalai/kernel/op_builder/utils.py
imports: import os
  import re
  import subprocess
  import warnings
  from typing import List
label_desc: Use image node_id 9 for calling a global function with example usage: get_cuda_cc_flag() and returns: cc_flag