common: updated MapSMtoCores for SM 5.3 to SM 7.0 (cdcseacave#489)

anapupa · Oct 18, 2019 · 22372e4 · 22372e4
1 parent ea15e14
commit 22372e4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/libs/Common/CUDA.cpp b/libs/Common/CUDA.cpp
@@ -42,6 +42,11 @@ int _convertSMVer2Cores(int major, int minor)
 		{0x37, 192}, // Kepler Generation (SM 3.7) GK21x class
 		{0x50, 128}, // Maxwell Generation (SM 5.0) GM10x class
 		{0x52, 128}, // Maxwell Generation (SM 5.2) GM20x class
+		{0x53, 128}, // Maxwell Generation (SM 5.3) GM20x class
+        	{0x60, 64 }, // Pascal Generation (SM 6.0) GP100 class
+        	{0x61, 128}, // Pascal Generation (SM 6.1) GP10x class
+        	{0x62, 128}, // Pascal Generation (SM 6.2) GP10x class
+        	{0x70, 64 }, // Volta Generation (SM 7.0) GV100 class
 		{-1, -1}
 	};