Allow H100

Author: williamwen42

intermediate_source/torch_compile_tutorial.py

@@ -35,7 +35,7 @@
 # - ``tabulate``
 
 ######################################################################
-# NOTE: a modern NVIDIA GPU (A100 or V100) is recommended for this tutorial in
+# NOTE: a modern NVIDIA GPU (H100, A100, or V100) is recommended for this tutorial in
 # order to reproduce the speedup numbers shown below and documented elsewhere.
 
 import torch
@@ -44,13 +44,13 @@
 gpu_ok = False
 if torch.cuda.is_available():
     device_cap = torch.cuda.get_device_capability()
-    if device_cap in [(7, 0), (8,0)]:
+    if device_cap in ((7, 0), (8, 0), (9, 0)):
         gpu_ok = True
 
 if not gpu_ok:
     warnings.warn(
-        "GPU is not NVIDIA V100 or A100. Speedup numbers may be lower than "
-        "expected."
+        "GPU is not NVIDIA V100, A100, or H100. Speedup numbers may be lower "
+        "than expected."
     )
 
 ######################################################################