accelerators.py 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. NVIDIA_TESLA_V100 = "V100"
  2. NVIDIA_TESLA_P100 = "P100"
  3. NVIDIA_TESLA_T4 = "T4"
  4. NVIDIA_TESLA_P4 = "P4"
  5. NVIDIA_TESLA_K80 = "K80"
  6. NVIDIA_TESLA_A10G = "A10G"
  7. NVIDIA_L4 = "L4"
  8. NVIDIA_L40S = "L40S"
  9. NVIDIA_A100 = "A100"
  10. NVIDIA_H100 = "H100"
  11. NVIDIA_H200 = "H200"
  12. NVIDIA_H20 = "H20"
  13. NVIDIA_B200 = "B200"
  14. INTEL_MAX_1550 = "Intel-GPU-Max-1550"
  15. INTEL_MAX_1100 = "Intel-GPU-Max-1100"
  16. INTEL_GAUDI = "Intel-GAUDI"
  17. AMD_INSTINCT_MI100 = "AMD-Instinct-MI100"
  18. AMD_INSTINCT_MI250x = "AMD-Instinct-MI250X"
  19. AMD_INSTINCT_MI250 = "AMD-Instinct-MI250X-MI250"
  20. AMD_INSTINCT_MI210 = "AMD-Instinct-MI210"
  21. AMD_INSTINCT_MI300A = "AMD-Instinct-MI300A"
  22. AMD_INSTINCT_MI300x = "AMD-Instinct-MI300X-OAM"
  23. AMD_INSTINCT_MI300x_HF = "AMD-Instinct-MI300X-HF"
  24. AMD_INSTINCT_MI308x = "AMD-Instinct-MI308X"
  25. AMD_INSTINCT_MI325x = "AMD-Instinct-MI325X-OAM"
  26. AMD_INSTINCT_MI350x = "AMD-Instinct-MI350X-OAM"
  27. AMD_INSTINCT_MI355x = "AMD-Instinct-MI355X-OAM"
  28. AMD_RADEON_R9_200_HD_7900 = "AMD-Radeon-R9-200-HD-7900"
  29. AMD_RADEON_HD_7900 = "AMD-Radeon-HD-7900"
  30. AWS_NEURON_CORE = "aws-neuron-core"
  31. GOOGLE_TPU_V2 = "TPU-V2"
  32. GOOGLE_TPU_V3 = "TPU-V3"
  33. GOOGLE_TPU_V4 = "TPU-V4"
  34. GOOGLE_TPU_V5P = "TPU-V5P"
  35. GOOGLE_TPU_V5LITEPOD = "TPU-V5LITEPOD"
  36. GOOGLE_TPU_V6E = "TPU-V6E"
  37. GOOGLE_TPU_V7X = "TPU-V7X"
  38. HUAWEI_NPU_910B = "Ascend910B"
  39. HUAWEI_NPU_910B4 = "Ascend910B4"
  40. METAX_C500 = "MXC500"
  41. METAX_C550 = "MXC550"
  42. # Use these instead of NVIDIA_A100 if you need a specific accelerator size. Note that
  43. # these labels are not auto-added to nodes, you'll have to add them manually in
  44. # addition to the default A100 label if needed.
  45. NVIDIA_A100_40G = "A100-40G"
  46. NVIDIA_A100_80G = "A100-80G"