| #ifndef THC_GENERAL_INC | |
| #define THC_GENERAL_INC | |
| #include "THGeneral.h" | |
| #include "cuda.h" | |
| #include "cublas.h" | |
| //#include "cuda_runtime_api.h" | |
| TH_API void THCudaInit(void); | |
| TH_API void THCudaShutdown(void); | |
| #define THCudaCheck(err) __THCudaCheck(err, __FILE__, __LINE__) | |
| #define THCublasCheck() __THCublasCheck( __FILE__, __LINE__) | |
| TH_API void __THCudaCheck(cudaError_t err, const char *file, const int line); | |
| TH_API void __THCublasCheck(const char *file, const int line); | |
| TH_API void THCudaGetGridSize(int *nBlockPerColumn_, int *nBlockPerRow_, int *nThreadPerBlock_, long size); | |
| #endif |