"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "88295f92d963f414cc7adf93f30c694a4d100dd2"
Unverified Commit 327589c8 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Polish the graphbolt code for labor sampling. (#6038)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent 69a532c1
...@@ -517,10 +517,12 @@ torch::Tensor Pick<SamplerType::LABOR>( ...@@ -517,10 +517,12 @@ torch::Tensor Pick<SamplerType::LABOR>(
return picked_neighbors; return picked_neighbors;
} else if (replace) { } else if (replace) {
return LaborPick<false, true>( return LaborPick<false, true>(
offset, num_neighbors, fanout, options, probs_or_mask, args); offset, num_neighbors, fanout, options,
/* probs_or_mask= */ torch::nullopt, args);
} else { // replace = false } else { // replace = false
return LaborPick<false, false>( return LaborPick<false, false>(
offset, num_neighbors, fanout, options, probs_or_mask, args); offset, num_neighbors, fanout, options,
/* probs_or_mask= */ torch::nullopt, args);
} }
} }
......
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