chore: import upstream snapshot with attribution

2026-07-13 13:17:40 +08:00
commit f1825c8ceb
10096 changed files with 2364182 additions and 0 deletions
@@ -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}")