fml  0.1-0
Fused Matrix Library
types.hh
1 // This file is part of fml which is released under the Boost Software
2 // License, Version 1.0. See accompanying file LICENSE or copy at
3 // https://www.boost.org/LICENSE_1_0.txt
4 
5 #ifndef FML_GPU_INTERNALS_CUDA_TYPES_H
6 #define FML_GPU_INTERNALS_CUDA_TYPES_H
7 #pragma once
8 
9 
10 #include <cublas.h>
11 #include <cuda_runtime.h>
12 #include <cusolverDn.h>
13 
14 typedef cudaError_t GPUError_t;
15 #define GPU_SUCCESS cudaSuccess
16 
17 #define GPUBLAS_STATUS_SUCCESS CUBLAS_STATUS_SUCCESS
18 typedef cublasStatus_t BLASStatus_t;
19 typedef cublasHandle_t BLASHandle_t;
20 
21 #define GPULAPACK_STATUS_SUCCESS CUSOLVER_STATUS_SUCCESS
22 typedef cusolverStatus_t LAPACKStatus_t;
23 typedef cusolverDnHandle_t LAPACKHandle_t;
24 
25 
26 #endif