#include <cstdio>
#include <cuda_runtime.h>
#include <cublas_v2.h>
#include <cufft.h>
#include <vector>
Functions | |
bool | gpuInit (FILE *fpLog=stdout, const class MPIUtil *mpiHostGpu=0, double *nGPUs=0) |
bool | isGpuMine () |
void | gpuErrorCheck () |
Check for gpu errors and print a useful message (implemented in GpuUtils.cpp) More... | |
bool | isGpuEnabled () |
A utility function to eliminate ugly #ifdef's when not required. | |
Variables | |
cublasHandle_t | cublasHandle |
global handle to cublas (defined in GpuUtil.cpp) | |