-
Notifications
You must be signed in to change notification settings - Fork 3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[GraphBolt] Implement dependent minibatching for labor. #7205
Conversation
To trigger regression tests:
|
@dgl-bot run g4dn.8xlarge model,api dmlc/master |
您好,信已收到,尽快给您回复。
|
@frozenbugs How do I run the regression tests to ensure this PR is not slowing down the CPU layer_neighbor implementation? |
Just merge the code, and the result will be send out tomorrow to you. If any unexpected happens, we can revert this PR. |
Description
The continuous seed file is mostly a copy of https://github.com/dmlc/dgl/blob/master/src/random/continuous_seed.h.
These changes are needed to implement the layer_dependency and batch_dependency arguments in https://docs.dgl.ai/en/latest/generated/dgl.dataloading.LaborSampler.html#dgl.dataloading.LaborSampler. They will be enabled in a future PR.
Checklist
Please feel free to remove inapplicable items for your PR.
Changes