[Core] Adding accelerator type H100 (#46823)

Signed-off-by: Jiajun Yao <[email protected]>
ray-project · Jul 27, 2024 · 2ac8712 · 2ac8712
1 parent 92acf8e
commit 2ac8712
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/python/ray/util/accelerators/__init__.py b/python/ray/util/accelerators/__init__.py
@@ -10,6 +10,7 @@
     NVIDIA_TESLA_A10G,
     NVIDIA_L4,
     NVIDIA_A100,
+    NVIDIA_H100,
     INTEL_MAX_1550,
     INTEL_MAX_1100,
     INTEL_GAUDI,
@@ -40,6 +41,7 @@
     "NVIDIA_A100",
     "NVIDIA_A100_40G",
     "NVIDIA_A100_80G",
+    "NVIDIA_H100",
     "INTEL_MAX_1550",
     "INTEL_MAX_1100",
     "INTEL_GAUDI",

diff --git a/python/ray/util/accelerators/accelerators.py b/python/ray/util/accelerators/accelerators.py
@@ -6,6 +6,7 @@
 NVIDIA_TESLA_A10G = "A10G"
 NVIDIA_L4 = "L4"
 NVIDIA_A100 = "A100"
+NVIDIA_H100 = "H100"
 INTEL_MAX_1550 = "Intel-GPU-Max-1550"
 INTEL_MAX_1100 = "Intel-GPU-Max-1100"
 INTEL_GAUDI = "Intel-GAUDI"