5 #ifndef FML_GPU_INTERNALS_CUDA_GPUPRIMS_H 6 #define FML_GPU_INTERNALS_CUDA_GPUPRIMS_H 11 #include <cusolverDn.h> 16 inline cudaError_t gpu_device_reset()
18 return cudaDeviceReset();
21 inline std::string gpu_error_string(cudaError_t code)
23 return cudaGetErrorString(code);
28 inline cublasStatus_t gpu_blas_init(cublasHandle_t *handle)
30 return cublasCreate(handle);
33 inline cublasStatus_t gpu_blas_free(cublasHandle_t handle)
35 return cublasDestroy(handle);
40 inline cusolverStatus_t gpu_lapack_init(cusolverDnHandle_t *handle)
42 return cusolverDnCreate(handle);
45 inline cusolverStatus_t gpu_lapack_free(cusolverDnHandle_t handle)
47 return cusolverDnDestroy(handle);
Definition: gpuprims.hh:14