diff --git a/ldgm/src/gpu.cu b/ldgm/src/gpu.cu index e9f0a87cb..18225780b 100644 --- a/ldgm/src/gpu.cu +++ b/ldgm/src/gpu.cu @@ -23,7 +23,6 @@ #include //#include "timer-util.h" #include "gpu.cuh" -#include "ldgm-session-gpu.h" struct coding_params { int num_lost; diff --git a/ldgm/src/gpu.cuh b/ldgm/src/gpu.cuh index e68dbbba6..6c63cf1be 100644 --- a/ldgm/src/gpu.cuh +++ b/ldgm/src/gpu.cuh @@ -1,3 +1,16 @@ +#include + +// CUDA check error +#define cuda_check_error(msg) \ + { \ + cudaError_t err = cudaGetLastError(); \ + if( cudaSuccess != err) { \ + fprintf(stderr, "[LDGM GPU] [Error] %s (line %i): %s: %s.\n", \ + __FILE__, __LINE__, msg, cudaGetErrorString( err) ); \ + exit(-1); \ + } \ + } \ + #ifdef __cplusplus extern "C" { #endif diff --git a/ldgm/src/ldgm-session-gpu.h b/ldgm/src/ldgm-session-gpu.h index de416a4ef..a0a4a303e 100644 --- a/ldgm/src/ldgm-session-gpu.h +++ b/ldgm/src/ldgm-session-gpu.h @@ -21,23 +21,11 @@ #define LDGM_SESSION_GPU_INC #include -#include #include "ldgm-session.h" #include #include -// CUDA check error -#define cuda_check_error(msg) \ - { \ - cudaError_t err = cudaGetLastError(); \ - if( cudaSuccess != err) { \ - fprintf(stderr, "[LDGM GPU] [Error] %s (line %i): %s: %s.\n", \ - __FILE__, __LINE__, msg, cudaGetErrorString( err) ); \ - exit(-1); \ - } \ - } \ - /* * ===================================================================================== * Class: LDGM_session_gpu