From 2ac8712497d287d150daa9e02de821ae409eaeb2 Mon Sep 17 00:00:00 2001
From: Jiajun Yao <jeromeyjj@gmail.com>
Date: Fri, 26 Jul 2024 22:08:10 -0700
Subject: [PATCH] [Core] Adding accelerator type H100 (#46823)

Signed-off-by: Jiajun Yao <jeromeyjj@gmail.com>
---
 python/ray/util/accelerators/__init__.py     | 2 ++
 python/ray/util/accelerators/accelerators.py | 1 +
 2 files changed, 3 insertions(+)

diff --git a/python/ray/util/accelerators/__init__.py b/python/ray/util/accelerators/__init__.py
index 8537f977752c..641cfc88508f 100644
--- 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
index 209eb3d5e73d..a572f98690dd 100644
--- 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"