Explorar o código

Fix new CUDA10 compilation errors (#4635)

FantasyGmm %!s(int64=2) %!d(string=hai) anos
pai
achega
77465dad48
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  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