common: add support for new GPU architectures in CUDA core conversion

cdcseacave · Dec 29, 2024 · 6ddbda1 · 6ddbda1
1 parent 3e0674a
commit 6ddbda1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/libs/Common/UtilCUDA.cpp b/libs/Common/UtilCUDA.cpp
@@ -51,6 +51,9 @@ int _convertSMVer2Cores(int major, int minor)
 		{0x72, 64 }, // Volta Generation (SM 7.2) GV10B class
 		{0x75, 64 }, // Turing Generation (SM 7.5) TU1xx class
 		{0x80, 64 }, // Ampere Generation (SM 8.0) GA100 class
+		{0x86, 128}, // Ampere Generation (SM 8.6) GA102-GA107 class
+		{0x89, 128}, // Ada Generation (SM 8.9) AD106 class
+		{0x90, 128}, // Hopper Generation (SM 9.0) GH100 class
 		{-1, -1}
 	};