From f42312e0a19e4dc5d7e332678bb451a7e9fb0cb3 Mon Sep 17 00:00:00 2001 From: slaren Date: Wed, 8 May 2024 16:38:48 +0200 Subject: [PATCH] replace minimum cc value with a constant --- ggml-cuda.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ggml-cuda.cu b/ggml-cuda.cu index d01797e0c..6f89a7cc3 100644 --- a/ggml-cuda.cu +++ b/ggml-cuda.cu @@ -2474,7 +2474,7 @@ GGML_CALL static enum ggml_status ggml_backend_cuda_graph_compute(ggml_backend_t void * ggml_cuda_cpy_fn_ptr = nullptr; if (cuda_ctx->cuda_graph->graph == nullptr) { - if (ggml_cuda_info().devices[cuda_ctx->device].cc < 800) { + if (ggml_cuda_info().devices[cuda_ctx->device].cc < CC_AMPERE) { cuda_ctx->cuda_graph->disable_due_to_gpu_arch = true; #ifndef NDEBUG fprintf(stderr, "%s: disabling CUDA graphs due to GPU architecture\n", __func__);