Browse Source

Tests: Add missing error check in CudaOnly.GPUDebugFlag

If `cudaMallocManaged` fails then later use of `has_debug` is not valid.
Brad King 8 years ago
parent
commit
8285aa9e4e
1 changed files with 5 additions and 0 deletions
  1. 5 0
      Tests/CudaOnly/GPUDebugFlag/main.cu

+ 5 - 0
Tests/CudaOnly/GPUDebugFlag/main.cu

@@ -49,6 +49,11 @@ int main(int argc, char** argv)
 {
   bool* has_debug;
   cudaError_t err = cudaMallocManaged(&has_debug, sizeof(bool));
+  if (err != cudaSuccess) {
+    std::cerr << "cudaMallocManaged failed:\n"
+              << "  " << cudaGetErrorString(err) << std::endl;
+    return 1;
+  }
 
   debug_kernel<<<1, 1>>>(has_debug);
   err = cudaDeviceSynchronize();