Merge pull request #16 from XSShawnZeng/master

Adding support for Blackwell
NVIDIA · Feb 5, 2025 · 9d03b03 · 9d03b03
2 parents e8a041d + 78f83ca
commit 9d03b03
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git 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
@@ -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_
-