Use RAFT_CUDA_TRY for cudaMemGetInfo.

bdice · bdice · commit 24d30362c139 · 2026-04-19T11:52:59.000-05:00
diff --git a/cpp/src/utilities/cuda_helpers.cuh b/cpp/src/utilities/cuda_helpers.cuh
@@ -241,7 +241,8 @@ DI void sorted_insert(T* array, T item, int curr_size, int max_size)
 inline size_t get_device_memory_size()
 {
   size_t free_mem, total_mem;
-  cudaMemGetInfo(&free_mem, &total_mem);
+  RAFT_CUDA_TRY(cudaMemGetInfo(&free_mem, &total_mem));
+  // TODO (bdice): Restore limiting adaptor check after updating CCCL to support resource_cast
   return total_mem;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -241,7 +241,8 @@ DI void sorted_insert(T* array, T item, int curr_size, int max_size)`
`241`	`241`	`inline size_t get_device_memory_size()`
`242`	`242`	`{`
`243`	`243`	`size_t free_mem, total_mem;`
`244`		`- cudaMemGetInfo(&free_mem, &total_mem);`
	`244`	`+ RAFT_CUDA_TRY(cudaMemGetInfo(&free_mem, &total_mem));`
	`245`	`+ // TODO (bdice): Restore limiting adaptor check after updating CCCL to support resource_cast`
`245`	`246`	`return total_mem;`
`246`	`247`	`}`
`247`	`248`