Просмотр исходного кода

Fixed CUDA runtime version check (#1879)

Johannes Gäßler 2 лет назад
Родитель
Сommit
a09f9195be
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      ggml-cuda.cu

+ 1 - 1
ggml-cuda.cu

@@ -25,7 +25,7 @@ static_assert(sizeof(half) == sizeof(ggml_fp16_t), "wrong fp16 size");
         }                                                                               \
     } while (0)
 
-#if CUDART_VERSION >= 12
+#if CUDART_VERSION >= 12000
 #define CUBLAS_CHECK(err)                                                               \
     do {                                                                                \
         cublasStatus_t err_ = (err);                                                    \