From 8285aa9e4e2df09b35f4faa930da6577979d4d2a Mon Sep 17 00:00:00 2001
From: Brad King <brad.king@kitware.com>
Date: Thu, 1 Feb 2018 09:07:50 -0500
Subject: Tests: Add missing error check in CudaOnly.GPUDebugFlag

If `cudaMallocManaged` fails then later use of `has_debug` is not valid.
---
 Tests/CudaOnly/GPUDebugFlag/main.cu | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/Tests/CudaOnly/GPUDebugFlag/main.cu b/Tests/CudaOnly/GPUDebugFlag/main.cu
index 8b97a3f..1f3fc12 100644
--- a/Tests/CudaOnly/GPUDebugFlag/main.cu
+++ b/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();
-- 
cgit v0.12