pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-03 19:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

change based on #7462

Browse Source
This commit is contained in:
Stanley Yang
2024-09-10 10:11:54 +00:00
parent bf125d8015
commit 1fc5f8a923
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
graphbolt/src/fused_csc_sampling_graph.cc
View File

@@ -1412,7 +1412,7 @@ inline int64_t UniformPick(
/** @brief An operator to perform non-uniform sampling. */
static torch::Tensor NonUniformPickOp(
torch::Tensor probs, int64_t fanout, bool replace) {
auto positive_probs_indices = probs.nonzero().squeeze(1);
auto positive_probs_indices = probs.contiguous().nonzero();
auto num_positive_probs = positive_probs_indices.size(0);
if (num_positive_probs == 0) return torch::empty({0}, torch::kLong);
if ((fanout == -1) || (num_positive_probs <= fanout && !replace)) {