@@ -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