浏览代码

Merge topic 'test-cuda-missing-check'

8285aa9e Tests: Add missing error check in CudaOnly.GPUDebugFlag

Acked-by: Kitware Robot <[email protected]>
Acked-by: Robert Maynard <[email protected]>
Merge-request: !1739
Brad King 7 年之前
父节点
当前提交
62c5fc80b1
共有 1 个文件被更改,包括 5 次插入0 次删除
  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();