File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -232,14 +232,14 @@ static bool fp16_mma_available(const int cc) {
232232 return false ;
233233#else
234234 return cc < GGML_CUDA_CC_OFFSET_AMD && ggml_cuda_highest_compiled_arch (cc) >= GGML_CUDA_CC_VOLTA ||
235- cc >= GGML_CUDA_CC_CDNA && cc < GGML_CUDA_CC_RDNA1 || cc >= GGML_CUDA_CC_RDNA3;
235+ GGML_CUDA_CC_IS_CDNA (cc) || cc >= GGML_CUDA_CC_RDNA3;
236236#endif // defined(GGML_USE_HIP) && defined(__HIP_PLATFORM_AMD__) && !defined(GGML_HIP_ROCWMMA_FATTN)
237237}
238238
239239// To be used for feature selection of external libraries, e.g. cuBLAS.
240240static bool fp16_mma_hardware_available (const int cc) {
241241 return cc < GGML_CUDA_CC_OFFSET_AMD && cc >= GGML_CUDA_CC_VOLTA ||
242- cc >= GGML_CUDA_CC_CDNA && cc < GGML_CUDA_CC_RDNA1 || cc >= GGML_CUDA_CC_RDNA3;
242+ GGML_CUDA_CC_IS_CDNA (cc) || cc >= GGML_CUDA_CC_RDNA3;
243243}
244244
245245// Volta technically had FP16 tensor cores but they work very differently compared to Turing and later.
You can’t perform that action at this time.
0 commit comments