"...utils/git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "2d8791324db0a764ff7088ba63741ad4f801abb4"
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(
offset, num_neighbors, fanouts, replace, options,
type_per_edge.value(), probs_or_mask, args, picked_data_ptr);
} else {
return Pick(
int64_t num_sampled = Pick(
offset, num_neighbors, fanouts[0], replace, options, probs_or_mask,
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