We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6a7220f commit 387206aCopy full SHA for 387206a
cpp/src/routing/utilities/cython.cu
@@ -113,7 +113,7 @@ std::vector<std::unique_ptr<vehicle_routing_ret_t>> call_batch_solve(
113
#pragma omp parallel for num_threads(max_thread)
114
for (std::size_t i = 0; i < size; ++i) {
115
// Required in multi-GPU environments to set the device for each thread
116
- cudaSetDevice(device_id);
+ RAFT_CUDA_TRY(cudaSetDevice(device_id));
117
118
auto old_stream = data_models[i]->get_handle_ptr()->get_stream();
119
// Make sure previous operations are finished
0 commit comments