Unverified Commit 23649071 authored by yxy235's avatar yxy235 Committed by GitHub
Browse files

[GraphBolt] Add sort to `Pick` in sampling. (#6571)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
parent e82f1746
...@@ -300,9 +300,13 @@ auto GetPickFn( ...@@ -300,9 +300,13 @@ auto GetPickFn(
offset, num_neighbors, fanouts, replace, options, offset, num_neighbors, fanouts, replace, options,
type_per_edge.value(), probs_or_mask, args, picked_data_ptr); type_per_edge.value(), probs_or_mask, args, picked_data_ptr);
} else { } else {
return Pick( int64_t num_sampled = Pick(
offset, num_neighbors, fanouts[0], replace, options, probs_or_mask, offset, num_neighbors, fanouts[0], replace, options, probs_or_mask,
args, picked_data_ptr); args, picked_data_ptr);
if (type_per_edge) {
std::sort(picked_data_ptr, picked_data_ptr + num_sampled);
}
return num_sampled;
} }
}; };
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment