5 #ifndef FML_GPU_INTERNALS_CUDA_TYPES_H 6 #define FML_GPU_INTERNALS_CUDA_TYPES_H 11 #include <cuda_runtime.h> 12 #include <cusolverDn.h> 14 typedef cudaError_t GPUError_t;
15 #define GPU_SUCCESS cudaSuccess 17 #define GPUBLAS_STATUS_SUCCESS CUBLAS_STATUS_SUCCESS 18 typedef cublasStatus_t BLASStatus_t;
19 typedef cublasHandle_t BLASHandle_t;
21 #define GPULAPACK_STATUS_SUCCESS CUSOLVER_STATUS_SUCCESS 22 typedef cusolverStatus_t LAPACKStatus_t;
23 typedef cusolverDnHandle_t LAPACKHandle_t;