Skip to content

Commit 20208e9

Browse files
authored
Merge branch 'ggml-org:master' into master
2 parents 0fa2a81 + 25ff6f7 commit 20208e9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml/src/ggml-cuda/common.cuh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,11 +312,11 @@ static bool turing_mma_available(const int cc) {
312312
}
313313

314314
static bool ampere_mma_available(const int cc) {
315-
return cc < GGML_CUDA_CC_OFFSET_AMD && ggml_cuda_highest_compiled_arch(cc) >= GGML_CUDA_CC_AMPERE;
315+
return GGML_CUDA_CC_IS_NVIDIA(cc) && ggml_cuda_highest_compiled_arch(cc) >= GGML_CUDA_CC_AMPERE;
316316
}
317317

318318
static bool cp_async_available(const int cc) {
319-
return cc < GGML_CUDA_CC_OFFSET_AMD && ggml_cuda_highest_compiled_arch(cc) >= GGML_CUDA_CC_AMPERE;
319+
return GGML_CUDA_CC_IS_NVIDIA(cc) && ggml_cuda_highest_compiled_arch(cc) >= GGML_CUDA_CC_AMPERE;
320320
}
321321

322322
static constexpr __device__ int ggml_cuda_get_physical_warp_size() {

0 commit comments

Comments
 (0)