gpu.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_GPU_H
#define FREE_TENSOR_GPU_H
 
#include <sstream>
 
#ifdef FT_WITH_CUDA
#include <cuda_runtime.h>
 
#include <except.h>
 
#define checkCudaError(call)                                                   \
    {                                                                          \
        auto err = (call);                                                     \
        if (cudaSuccess != err) {                                              \
            throw DriverError(cudaGetErrorString(err));                        \
        }                                                                      \
    }
 
#endif // FT_WITH_CUDA
 
#endif // FREE_TENSOR_GPU_H