fix missing GPUCHECK

MWATelescope · Jun 19, 2024 · e8414f3 · e8414f3
1 parent 7657753
commit e8414f3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/src/fee/gpu/fee.h b/src/fee/gpu/fee.h
@@ -434,15 +434,9 @@ extern "C" const char *gpu_fee_calc_jones(const FLOAT *d_azs, const FLOAT *d_zas
 
     gpuError_t error_id;
 #ifdef DEBUG
-    error_id = gpuDeviceSynchronize();
-    if (error_id != gpuSuccess) {
-        return gpuGetErrorString(error_id);
-    }
+    GPUCHECK(gpuDeviceSynchronize());
 #endif
-    error_id = gpuGetLastError();
-    if (error_id != gpuSuccess) {
-        return gpuGetErrorString(error_id);
-    }
+    GPUCHECK(gpuGetLastError());
 
     // Free device memory
     GPUCHECK(gpuFree(d_legendret));

diff --git a/src/gpu_common/gpu_common.cuh b/src/gpu_common/gpu_common.cuh
@@ -3,6 +3,8 @@
 #include <math.h>
 #include <stdint.h>
 #include <stdlib.h>
+#include <stdio.h>
+
 
 #ifdef SINGLE
 #define FLOAT  float
@@ -214,6 +216,18 @@ inline static __device__ FLOAT get_parallactic_angle(HADec hadec, FLOAT latitude
 
 #endif // BINDGEN
 
+#define GPUCHECK(error)                                               \
+    {                                                                 \
+        gpuError_t localError = error;                                \
+        if ((localError != gpuSuccess))                               \
+        {                                                             \
+            const char *errorString = gpuGetErrorString(localError);  \
+            printf("error: '%s'(%d) from %s at %s:%d\n", errorString, \
+                   localError, #error, __FILE__, __LINE__);           \
+            return errorString;                                       \
+        }                                                             \
+    }
+
 /*----------------------------------------------------------------------
 **
 **