Skip to content

Commit f7db90a

Browse files
committed
CUDA: Fix volta condition in ggml_cuda_should_use_wmma_fattn
1 parent d767dbe commit f7db90a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-cuda/fattn-wmma-f16.cuh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ static bool ggml_cuda_should_use_wmma_fattn(const int cc) {
2525
#if defined(GGML_USE_HIP) && !defined(GGML_HIP_ROCWMMA_FATTN)
2626
return false;
2727
#else
28-
if ((GGML_CUDA_CC_IS_NVIDIA(cc) && ggml_cuda_highest_compiled_arch(cc) >= GGML_CUDA_CC_VOLTA && cc == GGML_CUDA_CC_VOLTA) ||
28+
if ((GGML_CUDA_CC_IS_NVIDIA(cc) && ggml_cuda_highest_compiled_arch(cc) == GGML_CUDA_CC_VOLTA) ||
2929
GGML_CUDA_CC_IS_RDNA3(cc) || GGML_CUDA_CC_IS_MTHREADS(cc)) {
3030
return true;
3131
} else if (GGML_CUDA_CC_IS_CDNA(cc)){

0 commit comments

Comments
 (0)