Skip to content
Closed
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion cpp/src/traversal/od_shortest_distances_impl.cuh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION.
* SPDX-License-Identifier: Apache-2.0
*/
#pragma once
Expand Down Expand Up @@ -913,6 +913,7 @@ rmm::device_uvector<weight_t> od_shortest_distances(
handle.get_stream());

rmm::device_uvector<key_t> tmp_buffer = std::move(far_buffers.back());
far_buffers.back() = rmm::device_uvector<key_t>(0, handle.get_stream());
Comment thread
seunghwak marked this conversation as resolved.
Outdated
std::vector<key_t*> h_buffer_ptrs(h_split_thresholds.size() + 1);
auto old_size = new_near_q_keys.size();
for (size_t j = 0; j < h_buffer_ptrs.size(); ++j) {
Expand Down Expand Up @@ -995,6 +996,7 @@ rmm::device_uvector<weight_t> od_shortest_distances(
handle.get_stream());
} else {
far_buffers[i - num_near_q_insert_buffers] = std::move(far_buffers[i]);
far_buffers[i] = rmm::device_uvector<key_t>(0, handle.get_stream());
Comment thread
seunghwak marked this conversation as resolved.
Outdated
}
}

Expand Down
Loading