Skip to content

[BUG] Skip Empty Partitions in Bulk Sample Writing#3600

Merged
rapids-bot[bot] merged 4 commits intorapidsai:branch-23.06from
alexbarghi-nv:bulk-sampling-empty-partition-fix
May 24, 2023
Merged

[BUG] Skip Empty Partitions in Bulk Sample Writing#3600
rapids-bot[bot] merged 4 commits intorapidsai:branch-23.06from
alexbarghi-nv:bulk-sampling-empty-partition-fix

Conversation

@alexbarghi-nv
Copy link
Member

Currently, when the bulk sampler writes its samples, it does not check whether the partition is empty. This causes all of the empty partitions to collide and attempt to write to the same (empty) file at once. This writing should not occur at all.

This PR adds a check that skips writing if the partition is empty.

This should resolve the nightly test failures.

@alexbarghi-nv alexbarghi-nv self-assigned this May 23, 2023
@alexbarghi-nv alexbarghi-nv added bug Something isn't working non-breaking Non-breaking change labels May 23, 2023
@alexbarghi-nv alexbarghi-nv added this to the 23.06 milestone May 23, 2023
@alexbarghi-nv alexbarghi-nv marked this pull request as ready for review May 23, 2023 15:44
@alexbarghi-nv alexbarghi-nv requested a review from a team as a code owner May 23, 2023 15:44
Copy link
Member

@VibhuJawa VibhuJawa left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Thanks for triaging this @alexbarghi-nv

@alexbarghi-nv
Copy link
Member Author

/merge

@rapids-bot rapids-bot bot merged commit 28dc2eb into rapidsai:branch-23.06 May 24, 2023
nv-rliu pushed a commit to nv-rliu/cugraph-gnn that referenced this pull request Jul 15, 2025
Currently, when the bulk sampler writes its samples, it does not check whether the partition is empty.  This causes all of the empty partitions to collide and attempt to write to the same (empty) file at once.  This writing should not occur at all.

This PR adds a check that skips writing if the partition is empty.

This should resolve the nightly test failures.

Authors:
  - Alex Barghi (https://github.com/alexbarghi-nv)

Approvers:
  - Vibhu Jawa (https://github.com/VibhuJawa)
  - Brad Rees (https://github.com/BradReesWork)

URL: rapidsai/cugraph#3600
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants