From 229960a5e9995643ce0ce957a57d847effdc41dc Mon Sep 17 00:00:00 2001
From: Logan Adams <114770087+loadams@users.noreply.github.com>
Date: Sun, 27 Oct 2024 20:39:51 -0700
Subject: [PATCH] Add support for H100/sm_90 arch compilation (#6669)

Resolves: #6549
---
 op_builder/builder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/op_builder/builder.py b/op_builder/builder.py
index 1609bc9005f4..461281d4a569 100644
--- a/op_builder/builder.py
+++ b/op_builder/builder.py
@@ -67,7 +67,7 @@ def get_default_compute_capabilities():
             # Special treatment of CUDA 11.0 because compute_86 is not supported.
             compute_caps += ";8.0"
         else:
-            compute_caps += ";8.0;8.6"
+            compute_caps += ";8.0;8.6;9.0"
     return compute_caps