From 78f83ca02d1cc22b4f9146a195ce947ab6c75e7e Mon Sep 17 00:00:00 2001
From: shawnz <shawnz@nvidia.com>
Date: Wed, 5 Feb 2025 11:17:06 +0800
Subject: [PATCH] Adding suport for Blackwell

---
 Common/helper_cuda.h        | 6 ++++++
 Common/helper_cuda_drvapi.h | 4 +++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/Common/helper_cuda.h b/Common/helper_cuda.h
index 22d8eeaa..dc0efc46 100644
--- a/Common/helper_cuda.h
+++ b/Common/helper_cuda.h
@@ -668,6 +668,9 @@ inline int _ConvertSMVer2Cores(int major, int minor) {
       {0x87, 128},
       {0x89, 128},
       {0x90, 128},
+      {0xa0, 128},
+      {0xa1, 128},
+      {0xc0, 128},
       {-1, -1}};
 
   int index = 0;
@@ -717,6 +720,9 @@ inline const char* _ConvertSMVer2ArchName(int major, int minor) {
       {0x87, "Ampere"},
       {0x89, "Ada"},
       {0x90, "Hopper"},
+      {0xa0, "Blackwell"},
+      {0xa1, "Blackwell"},
+      {0xc0, "Blackwell"},
       {-1, "Graphics Device"}};
 
   int index = 0;
diff --git a/Common/helper_cuda_drvapi.h b/Common/helper_cuda_drvapi.h
index e746ce61..37326761 100644
--- a/Common/helper_cuda_drvapi.h
+++ b/Common/helper_cuda_drvapi.h
@@ -116,6 +116,9 @@ inline int _ConvertSMVer2CoresDRV(int major, int minor) {
       {0x87, 128},
       {0x89, 128},
       {0x90, 128},
+      {0xa0, 128},
+      {0xa1, 128},
+      {0xc0, 128},
       {-1, -1}};
 
   int index = 0;
@@ -405,4 +408,3 @@ bool inline findFatbinPath(const char *module_file, std::string &module_path, ch
   // end of CUDA Helper Functions
 
 #endif  // COMMON_HELPER_CUDA_DRVAPI_H_
-