From 1c3cb7f389347efc33abbffb9c30315211d4fa7a Mon Sep 17 00:00:00 2001 From: Rajesh Gandham Date: Fri, 27 Jun 2025 06:52:58 -0700 Subject: [PATCH 1/2] Do not pass dynamic shared memory size for extract_best_per_route kernel --- cpp/src/routing/ges/squeeze.cu | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cpp/src/routing/ges/squeeze.cu b/cpp/src/routing/ges/squeeze.cu index a8553b159..db9762eba 100644 --- a/cpp/src/routing/ges/squeeze.cu +++ b/cpp/src/routing/ges/squeeze.cu @@ -116,11 +116,11 @@ i_t guided_ejection_search_t::try_multiple_insert(i_t n_inser RAFT_CHECK_CUDA(solution_ptr->sol_handle->get_stream()); if constexpr (squeeze_mode) { - size_t move_blocks = solution_ptr->get_num_requests(); + i_t move_blocks = solution_ptr->get_num_requests(); extract_best_per_route - <<>>(solution_ptr->view(), - cuopt::make_span(best_squeeze_per_cand), - cuopt::make_span(best_squeeze_per_route)); + <<>>(solution_ptr->view(), + cuopt::make_span(best_squeeze_per_cand), + cuopt::make_span(best_squeeze_per_route)); RAFT_CHECK_CUDA(solution_ptr->sol_handle->get_stream()); } From e5ff891ae7af631abb066c228ec166e88aa05245 Mon Sep 17 00:00:00 2001 From: Rajesh Gandham Date: Fri, 27 Jun 2025 07:16:23 -0700 Subject: [PATCH 2/2] Revert unnecessary change --- cpp/src/routing/ges/squeeze.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/src/routing/ges/squeeze.cu b/cpp/src/routing/ges/squeeze.cu index db9762eba..1e27545a6 100644 --- a/cpp/src/routing/ges/squeeze.cu +++ b/cpp/src/routing/ges/squeeze.cu @@ -116,7 +116,7 @@ i_t guided_ejection_search_t::try_multiple_insert(i_t n_inser RAFT_CHECK_CUDA(solution_ptr->sol_handle->get_stream()); if constexpr (squeeze_mode) { - i_t move_blocks = solution_ptr->get_num_requests(); + size_t move_blocks = solution_ptr->get_num_requests(); extract_best_per_route <<>>(solution_ptr->view(), cuopt::make_span(best_squeeze_per_cand),