From 8ad66c7a69e1f6672153a37e643d6a8cb8d7d77b Mon Sep 17 00:00:00 2001 From: Artem Chirkin <9253178+achirkin@users.noreply.github.com> Date: Tue, 11 Aug 2026 04:30:38 -0700 Subject: [PATCH 1/2] Fix lancsoz launch grid --- cpp/include/raft/sparse/solver/detail/lanczos.cuh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cpp/include/raft/sparse/solver/detail/lanczos.cuh b/cpp/include/raft/sparse/solver/detail/lanczos.cuh index 9aa4304c04..af58b1f1a0 100644 --- a/cpp/include/raft/sparse/solver/detail/lanczos.cuh +++ b/cpp/include/raft/sparse/solver/detail/lanczos.cuh @@ -1,5 +1,5 @@ /* - * SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION. + * SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. * SPDX-License-Identifier: Apache-2.0 */ @@ -157,13 +157,13 @@ void lanczos_solve_ritz( // alpha.data_handle(), triangular_matrix.data_handle(), ncv, ncv, stream); int blockSize = 256; - int numBlocks = (ncv + blockSize - 1) / blockSize; + int numBlocks = raft::div_rounding_up_safe(ncv, blockSize); kernel_triangular_populate - <<>>(triangular_matrix.data_handle(), beta.data_handle(), ncv); + <<>>(triangular_matrix.data_handle(), beta.data_handle(), ncv); if (beta_k) { int threadsPerBlock = 256; - int blocksPerGrid = (k + threadsPerBlock - 1) / threadsPerBlock; + int blocksPerGrid = raft::div_rounding_up_safe(k, threadsPerBlock); kernel_triangular_beta_k<<>>( triangular_matrix.data_handle(), beta_k.value().data_handle(), (int)k, ncv); } From 46b92521dbd6579bcf847c26f3d88bc430d2c511 Mon Sep 17 00:00:00 2001 From: Artem Chirkin <9253178+achirkin@users.noreply.github.com> Date: Tue, 11 Aug 2026 05:11:38 -0700 Subject: [PATCH 2/2] cast k to int --- cpp/include/raft/sparse/solver/detail/lanczos.cuh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/include/raft/sparse/solver/detail/lanczos.cuh b/cpp/include/raft/sparse/solver/detail/lanczos.cuh index af58b1f1a0..e33f6b6ef0 100644 --- a/cpp/include/raft/sparse/solver/detail/lanczos.cuh +++ b/cpp/include/raft/sparse/solver/detail/lanczos.cuh @@ -163,7 +163,7 @@ void lanczos_solve_ritz( if (beta_k) { int threadsPerBlock = 256; - int blocksPerGrid = raft::div_rounding_up_safe(k, threadsPerBlock); + int blocksPerGrid = raft::div_rounding_up_safe(k, threadsPerBlock); kernel_triangular_beta_k<<>>( triangular_matrix.data_handle(), beta_k.value().data_handle(), (int)k, ncv); }