replace minimum cc value with a constant

ggerganov · May 8, 2024 · f42312e · f42312e
1 parent ab40e66
commit f42312e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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__);