code: def get_gpu_archs(command):
      archs = []
      for loc in range(len(command)):
          if command[loc] != "-gencode":
              continue
          arch_flag = command[loc + 1]
          match = GPU_ARCH_REGEX.search(arch_flag)
          if match is not None:
              archs.append("--cuda-gpu-arch=sm_%s" % match.group(1))
      return archs
  
imports: import argparse
  import json
  import multiprocessing
  import os
  import re
  import shutil
  import subprocess
  import sys
  from pathlib import Path
  import tomli
label_desc: Use image node_id 4 for calling a global function with example usage: get_gpu_archs(command) and returns: archs