Skip to content

Commit 387206a

Browse files
committed
Add error checking
1 parent 6a7220f commit 387206a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cpp/src/routing/utilities/cython.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ std::vector<std::unique_ptr<vehicle_routing_ret_t>> call_batch_solve(
113113
#pragma omp parallel for num_threads(max_thread)
114114
for (std::size_t i = 0; i < size; ++i) {
115115
// Required in multi-GPU environments to set the device for each thread
116-
cudaSetDevice(device_id);
116+
RAFT_CUDA_TRY(cudaSetDevice(device_id));
117117

118118
auto old_stream = data_models[i]->get_handle_ptr()->get_stream();
119119
// Make sure previous operations are finished

0 commit comments

Comments
 (0)