code: def get_gpu_type() -> NVIDIA_GPU_TYPE:
      gpu_info = torch.utils.collect_env.get_gpu_info(
          torch.utils.collect_env.run
      )
      if "V100" in gpu_info:
          return NVIDIA_GPU_TYPE.VOLTA
      elif "A100" in gpu_info:
          return NVIDIA_GPU_TYPE.AMPERE
      elif "H100" in gpu_info:
          return NVIDIA_GPU_TYPE.HOPPER
      else:
          # for other gpu types, assume Ampere
          return NVIDIA_GPU_TYPE.AMPERE
  
file_path: pytorch/torch/_inductor/comm_analysis.py
imports: import math
  from enum import IntEnum
  from typing import TYPE_CHECKING
  import torch
  from .None import ir
  from .utils import get_dtype_size, sympy_product
  from .virtualized import V
label_desc: Use image node_id 1 for calling a global function with example usage: get_gpu_type() and returns: NVIDIA_GPU_TYPE, NVIDIA_GPU_TYPE, NVIDIA_GPU_TYPE, NVIDIA_GPU_TYPE