tAdd support for CUDA compute capabilities 4.x, 6.2, and 7.x - sphere - GPU-bas… | |
git clone git://src.adamsgaard.dk/sphere | |
Log | |
Files | |
Refs | |
LICENSE | |
--- | |
commit 116df3bcf276dec7fca4fc44a8b91c39db229f66 | |
parent 167ae1fe9a881f423cdffd2f3737ce6e9de69f78 | |
Author: Anders Damsgaard <[email protected]> | |
Date: Sat, 27 Apr 2019 14:27:26 +0200 | |
Add support for CUDA compute capabilities 4.x, 6.2, and 7.x | |
Diffstat: | |
M src/device.cu | 7 +++++++ | |
1 file changed, 7 insertions(+), 0 deletions(-) | |
--- | |
diff --git a/src/device.cu b/src/device.cu | |
t@@ -31,6 +31,7 @@ | |
// Returns the number of cores per streaming multiprocessor, which is | |
// a function of the device compute capability | |
+// https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#compute-ca… | |
int cudaCoresPerSM(int major, int minor) | |
{ | |
if (major == 1) | |
t@@ -41,12 +42,18 @@ int cudaCoresPerSM(int major, int minor) | |
return 48; | |
else if (major == 3) | |
return 192; | |
+ else if (major == 4) | |
+ return 128; | |
else if (major == 5) | |
return 128; | |
else if (major == 6 && minor == 0) | |
return 64; | |
else if (major == 6 && minor == 1) | |
return 128; | |
+ else if (major == 6 && minor == 2) | |
+ return 128; | |
+ else if (major == 7) | |
+ return 32; | |
else | |
printf("Error in cudaCoresPerSM Device compute capability value " | |
"(%d.%d) not recognized.", major, minor); |