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

Fix new CUDA10 compilation errors (#4635)

FantasyGmm 2 лет назад
Родитель
Сommit
77465dad48
1 измененных файлов с 1 добавлено и 0 удалено
  1. 1 0
      ggml-cuda.cu

+ 1 - 0
ggml-cuda.cu

@@ -102,6 +102,7 @@
 #include <cuda_fp16.h>
 
 #if CUDART_VERSION < 11020
+#define CU_DEVICE_ATTRIBUTE_VIRTUAL_MEMORY_MANAGEMENT_SUPPORTED CU_DEVICE_ATTRIBUTE_VIRTUAL_ADDRESS_MANAGEMENT_SUPPORTED
 #define CUBLAS_TF32_TENSOR_OP_MATH CUBLAS_TENSOR_OP_MATH
 #define CUBLAS_COMPUTE_16F CUDA_R_16F
 #define CUBLAS_COMPUTE_32F CUDA_R_32F