Skip to content

Commit f14874a

Browse files
committed
reapply fix from 5381 rather than merging after so many updates
1 parent 6bc8593 commit f14874a

File tree

1 file changed

+9
-17
lines changed

1 file changed

+9
-17
lines changed

cpp/src/sampling/detail/temporal_partition_vertices_impl.cuh

Lines changed: 9 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -141,28 +141,20 @@ temporal_partition_vertices(raft::handle_t const& handle,
141141
vertex_labels_p1->resize(vertices_p1.size(), handle.get_stream());
142142
vertex_times_p1.resize(vertices_p1.size(), handle.get_stream());
143143
} else {
144-
copy_if_mask_unset(
145-
handle,
146-
thrust::make_zip_iterator(
147-
vertices_p1.begin(), vertex_times_p1.begin(), vertex_labels_p1->begin()),
148-
thrust::make_zip_iterator(
149-
vertices_p1.end(), vertex_times_p1.end(), vertex_labels_p1->end()),
150-
vertex_partition_mask.begin(),
151-
thrust::make_zip_iterator(
152-
vertices_p2.begin(), vertex_times_p2.begin(), vertex_labels_p2->begin()));
144+
copy_if_mask_unset(handle,
145+
thrust::make_zip_iterator(vertices_p1.begin(), vertex_times_p1.begin()),
146+
thrust::make_zip_iterator(vertices_p1.end(), vertex_times_p1.end()),
147+
vertex_partition_mask.begin(),
148+
thrust::make_zip_iterator(vertices_p2.begin(), vertex_times_p2.begin()));
153149
vertices_p1.resize(
154150
cuda::std::distance(
155-
thrust::make_zip_iterator(
156-
vertices_p1.begin(), vertex_times_p1.begin(), vertex_labels_p1->begin()),
151+
thrust::make_zip_iterator(vertices_p1.begin(), vertex_times_p1.begin()),
157152
copy_if_mask_set(
158153
handle,
159-
thrust::make_zip_iterator(
160-
vertices_p1.begin(), vertex_times_p1.begin(), vertex_labels_p1->begin()),
161-
thrust::make_zip_iterator(
162-
vertices_p1.end(), vertex_times_p1.end(), vertex_labels_p1->end()),
154+
thrust::make_zip_iterator(vertices_p1.begin(), vertex_times_p1.begin()),
155+
thrust::make_zip_iterator(vertices_p1.end(), vertex_times_p1.end()),
163156
vertex_partition_mask.begin(),
164-
thrust::make_zip_iterator(
165-
vertices_p1.begin(), vertex_times_p1.begin(), vertex_labels_p1->begin()))),
157+
thrust::make_zip_iterator(vertices_p1.begin(), vertex_times_p1.begin()))),
166158
handle.get_stream());
167159

168160
vertex_times_p1.resize(vertices_p1.size(), handle.get_stream());

0 commit comments

Comments
 (0)