Use RAFT_CUDA_TRY for cudaMemGetInfo.

bdice · bdice · commit 8d387c2105ba · 2026-04-19T11:51:51.000-05:00
diff --git a/cpp/src/utilities/cuda_helpers.cuh b/cpp/src/utilities/cuda_helpers.cuh
@@ -241,7 +241,7 @@ 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,7 @@ 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`	`245`	`// TODO (bdice): Restore limiting adaptor check after updating CCCL to support resource_cast`
`246`	`246`	`return total_mem;`
`247`	`247`	`}`