diff options
| author | FantasyGmm <16450052+FantasyGmm@users.noreply.github.com> | 2023-12-26 18:38:36 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-26 11:38:36 +0100 |
| commit | 77465dad48d7c945c367ab46b6f2ea98ae9b7b15 (patch) | |
| tree | 83b936e9ca3f42d1ae76e20ab7161405986210dd /ggml-cuda.cu | |
| parent | a206137f927daef1752753cf5e281220b449a468 (diff) | |
Fix new CUDA10 compilation errors (#4635)
Diffstat (limited to 'ggml-cuda.cu')
| -rw-r--r-- | ggml-cuda.cu | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ggml-cuda.cu b/ggml-cuda.cu index ac3b3c14..f32e83ab 100644 --- a/ggml-cuda.cu +++ b/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 |