try to avoid fp16 for anything below cc 610

2026-09-18 16:55:14 +02:00 · 2025-03-05 17:52:59 +08:00
parent b80b784ee5
commit 2fdef19d1b
1 changed files with 1 additions and 1 deletions
@@ -218,7 +218,7 @@ static bool fp16_available(const int cc) {
 }

 static bool fast_fp16_available(const int cc) {
-    return fp16_available(cc) && cc != 610;
+    return fp16_available(cc) && cc > 610;
 }

 // To be used for feature selection of external libraries, e.g. cuBLAS.