chore: import upstream snapshot with attribution

python/ray/util/accelerators/__init__.py

@@ -0,0 +1,84 @@
+import warnings
+
+from ray.util import tpu
+from ray.util.accelerators.accelerators import (
+    AMD_INSTINCT_MI100,
+    AMD_INSTINCT_MI210,
+    AMD_INSTINCT_MI250,
+    AMD_RADEON_HD_7900,
+    AMD_RADEON_R9_200_HD_7900,
+    AWS_NEURON_CORE,
+    GOOGLE_TPU_V2,
+    GOOGLE_TPU_V3,
+    GOOGLE_TPU_V4,
+    GOOGLE_TPU_V5LITEPOD,
+    GOOGLE_TPU_V5P,
+    GOOGLE_TPU_V6E,
+    GOOGLE_TPU_V7X,
+    INTEL_GAUDI,
+    INTEL_MAX_1100,
+    INTEL_MAX_1550,
+    METAX_C500,
+    METAX_C550,
+    NVIDIA_A100,
+    NVIDIA_H100,
+    NVIDIA_L4,
+    NVIDIA_TESLA_A10G,
+    NVIDIA_TESLA_K80,
+    NVIDIA_TESLA_P4,
+    NVIDIA_TESLA_P100,
+    NVIDIA_TESLA_T4,
+    NVIDIA_TESLA_V100,
+    AMD_INSTINCT_MI250x,
+    AMD_INSTINCT_MI300x,
+)
+
+__all__ = [
+    "tpu",
+    "NVIDIA_TESLA_V100",
+    "NVIDIA_TESLA_P100",
+    "NVIDIA_TESLA_T4",
+    "NVIDIA_TESLA_P4",
+    "NVIDIA_TESLA_K80",
+    "NVIDIA_TESLA_A10G",
+    "NVIDIA_L4",
+    "NVIDIA_A100",
+    "NVIDIA_A100_40G",
+    "NVIDIA_A100_80G",
+    "NVIDIA_H100",
+    "INTEL_MAX_1550",
+    "INTEL_MAX_1100",
+    "INTEL_GAUDI",
+    "AMD_INSTINCT_MI100",
+    "AMD_INSTINCT_MI210",
+    "AMD_INSTINCT_MI250",
+    "AMD_INSTINCT_MI250x",
+    "AMD_INSTINCT_MI300x",
+    "AMD_RADEON_R9_200_HD_7900",
+    "AMD_RADEON_HD_7900",
+    "AWS_NEURON_CORE",
+    "GOOGLE_TPU_V2",
+    "GOOGLE_TPU_V3",
+    "GOOGLE_TPU_V4",
+    "GOOGLE_TPU_V5P",
+    "GOOGLE_TPU_V5LITEPOD",
+    "GOOGLE_TPU_V6E",
+    "GOOGLE_TPU_V7X",
+    "METAX_C500",
+    "METAX_C550",
+    # Deprecated
+    "NVIDIA_TESLA_A100",
+]
+
+
+def __getattr__(name: str):
+    if name == "NVIDIA_TESLA_A100":
+        from ray.util.annotations import RayDeprecationWarning
+
+        warnings.warn(
+            "NVIDIA_TESLA_A100 is deprecated, use NVIDIA_A100 instead.",
+            RayDeprecationWarning,
+            stacklevel=2,
+        )
+        return NVIDIA_A100
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

python/ray/util/accelerators/accelerators.py

@@ -0,0 +1,50 @@
+NVIDIA_TESLA_V100 = "V100"
+NVIDIA_TESLA_P100 = "P100"
+NVIDIA_TESLA_T4 = "T4"
+NVIDIA_TESLA_P4 = "P4"
+NVIDIA_TESLA_K80 = "K80"
+NVIDIA_TESLA_A10G = "A10G"
+NVIDIA_L4 = "L4"
+NVIDIA_L40S = "L40S"
+NVIDIA_A100 = "A100"
+NVIDIA_H100 = "H100"
+NVIDIA_H200 = "H200"
+NVIDIA_H20 = "H20"
+NVIDIA_B200 = "B200"
+NVIDIA_B300 = "B300"
+NVIDIA_RTX_PRO_6000 = "RTX-PRO-6000"
+INTEL_MAX_1550 = "Intel-GPU-Max-1550"
+INTEL_MAX_1100 = "Intel-GPU-Max-1100"
+INTEL_GAUDI = "Intel-GAUDI"
+AMD_INSTINCT_MI100 = "AMD-Instinct-MI100"
+AMD_INSTINCT_MI250x = "AMD-Instinct-MI250X"
+AMD_INSTINCT_MI250 = "AMD-Instinct-MI250X-MI250"
+AMD_INSTINCT_MI210 = "AMD-Instinct-MI210"
+AMD_INSTINCT_MI300A = "AMD-Instinct-MI300A"
+AMD_INSTINCT_MI300x = "AMD-Instinct-MI300X-OAM"
+AMD_INSTINCT_MI300x_HF = "AMD-Instinct-MI300X-HF"
+AMD_INSTINCT_MI308x = "AMD-Instinct-MI308X"
+AMD_INSTINCT_MI325x = "AMD-Instinct-MI325X-OAM"
+AMD_INSTINCT_MI350x = "AMD-Instinct-MI350X-OAM"
+AMD_INSTINCT_MI355x = "AMD-Instinct-MI355X-OAM"
+AMD_RADEON_R9_200_HD_7900 = "AMD-Radeon-R9-200-HD-7900"
+AMD_RADEON_HD_7900 = "AMD-Radeon-HD-7900"
+AWS_NEURON_CORE = "aws-neuron-core"
+GOOGLE_TPU_V2 = "TPU-V2"
+GOOGLE_TPU_V3 = "TPU-V3"
+GOOGLE_TPU_V4 = "TPU-V4"
+GOOGLE_TPU_V5P = "TPU-V5P"
+GOOGLE_TPU_V5LITEPOD = "TPU-V5LITEPOD"
+GOOGLE_TPU_V6E = "TPU-V6E"
+GOOGLE_TPU_V7X = "TPU-V7X"
+HUAWEI_NPU_910B = "Ascend910B"
+HUAWEI_NPU_910B4 = "Ascend910B4"
+METAX_C500 = "MXC500"
+METAX_C550 = "MXC550"
+FURIOSA_RNGD = "FURIOSA_RNGD"
+
+# Use these instead of NVIDIA_A100 if you need a specific accelerator size. Note that
+# these labels are not auto-added to nodes, you'll have to add them manually in
+# addition to the default A100 label if needed.
+NVIDIA_A100_40G = "A100-40G"
+NVIDIA_A100_80G = "A100-80G"