fix inaccurate core count on Volta, Turing and Pascal

fangq · Jul 29, 2020 · daa9d56 · daa9d56
1 parent b5d245f
commit daa9d56
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/mcx_core.cu b/src/mcx_core.cu
@@ -1563,7 +1563,8 @@ int mcx_corecount(int v1, int v2){
      else if(v<21) return 32;
      else if(v<30) return 48;
      else if(v<50) return 192;
-     else          return 128;
+     else if(v<60 || v==61) return 128;
+     else          return 64;
 }
 
 /**