[GraphBolt] `index_select_csc_batched`.

dmlc / dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

http://dgl.ai

Apache License 2.0

13.53k stars 3.02k forks source link

[GraphBolt] `index_select_csc_batched`. #7511

Closed mfbalin closed 4 months ago

mfbalin commented 4 months ago

Description

The overlap_graph_fetch code path benefits from this change by avoiding the duplicate indexing of the indptr tensor multiple times for each index_select_csc call..

Checklist

Please feel free to remove inapplicable items for your PR.

[ ] The PR title starts with [$CATEGORY] (such as [NN], [Model], [Doc], [Feature]])
[ ] I've leverage the tools to beautify the python and c++ code.
[ ] The PR is complete and small, read the Google eng practice (CL equals to PR) to understand more about small PR. In DGL, we consider PRs with less than 200 lines of core code change are small (example, test and documentation could be exempted).
[ ] All changes have test coverage
[ ] Code is well-documented
[ ] To the best of my knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change
[ ] Related issue is referred in this PR
[ ] If the PR is for a new model/paper, I've updated the example index here.

Changes

dgl-bot commented 4 months ago

To trigger regression tests:

@dgl-bot run [instance-type] [which tests] [compare-with-branch]; For example: @dgl-bot run g4dn.4xlarge all dmlc/master or @dgl-bot run c5.9xlarge kernel,api dmlc/master

dgl-bot commented 4 months ago

Commit ID: dc81d5d5d700fa5cbbdfeb73d696383f377862f1

Build ID: 1

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: ab777aa018cd30276f848a8ef7b1c3b9cd5ef5c8

Build ID: 2

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link

mfbalin commented 4 months ago

Adding tests in a bit.

mfbalin commented 4 months ago

@frozenbugs Appreciate the quick review.

dgl-bot commented 4 months ago

Commit ID: ca9e40a2e49c4dddf2520353d6c5a03f1c87dc93

Build ID: 3

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: 1f044accbcaca98e0506eff0f263e6de2ad87551

Build ID: 4

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: 0055509d89bf167e61e96ba0c02b9f6cfba8d86c

Build ID: 5

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: 015ba8dbe15a8f49e14c64a72d86583978d46504

Build ID: 6

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: f5c0eaf244053e743fad5a80ad7959f84e7ef1eb

Build ID: 7

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 4 months ago

Commit ID: c0a4e7018ac1a4e61797a5e4586ef0c9ef93c211

Build ID: 8

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link