From cda319790f18bf921e46ec8a0c48529eb359e2ac Mon Sep 17 00:00:00 2001 From: Chuck Hastings Date: Sat, 26 Feb 2022 18:11:14 -0500 Subject: [PATCH] if batchsize > 1, the dualUpdate function was not allocating sufficient space for all of the answers --- cpp/include/raft/lap/detail/lap_functions.cuh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/include/raft/lap/detail/lap_functions.cuh b/cpp/include/raft/lap/detail/lap_functions.cuh index 3a801ff060..e03185f508 100644 --- a/cpp/include/raft/lap/detail/lap_functions.cuh +++ b/cpp/include/raft/lap/detail/lap_functions.cuh @@ -466,7 +466,7 @@ inline void dualUpdate(raft::handle_t const& handle, dim3 threads_per_block; int total_blocks; - rmm::device_scalar sp_min_v(handle.get_stream()); + rmm::device_uvector sp_min_v(SP, handle.get_stream()); raft::lap::detail::calculateLinearDims(blocks_per_grid, threads_per_block, total_blocks, SP); kernel_dualUpdate_1<<>>(