Parcourir la source

Fix new CUDA10 compilation errors (#4635)

FantasyGmm il y a 2 ans
Parent
commit
77465dad48
1 fichiers modifiés avec 1 ajouts et 0 suppressions
  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