Actually throw bad_alloc from cuda_cub::malloc.

brycelelbach · brycelelbach · commit 5ee2f881684f · 2020-02-03T10:27:49.000-08:00
Bug 2813126
Bug 2808654

Reviewed-by: Michał 'Griwes' Dominiak &lt;griwes@griwes.info&gt;
diff --git a/thrust/system/cuda/detail/malloc_and_free.h b/thrust/system/cuda/detail/malloc_and_free.h
@@ -63,7 +63,7 @@ void *malloc(execution_policy<DerivedPolicy> &, std::size_t n)
   if(status != cudaSuccess)
   {
     cudaGetLastError(); // Clear global CUDA error state.
-    thrust::system::detail::bad_alloc(thrust::cuda_category().message(status).c_str());
+    throw thrust::system::detail::bad_alloc(thrust::cuda_category().message(status).c_str());
   }
 #else
   result = thrust::raw_pointer_cast(thrust::malloc(thrust::seq, n));

Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,7 @@ void *malloc(execution_policy<DerivedPolicy> &, std::size_t n)`
`63`	`63`	`if(status != cudaSuccess)`
`64`	`64`	`{`
`65`	`65`	`cudaGetLastError(); // Clear global CUDA error state.`
`66`		`- thrust::system::detail::bad_alloc(thrust::cuda_category().message(status).c_str());`
	`66`	`+ throw thrust::system::detail::bad_alloc(thrust::cuda_category().message(status).c_str());`
`67`	`67`	`}`
`68`	`68`	`#else`
`69`	`69`	`result = thrust::raw_pointer_cast(thrust::malloc(thrust::seq, n));`