Fix new CUDA10 compilation errors (#4635)

This commit is contained in:
FantasyGmm
2023-12-26 18:38:36 +08:00
committed by GitHub
parent a206137f92
commit 77465dad48

View File

@@ -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