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

The CreateBlock is slow after we optimized GB dataloader. #7412

Open frozenbugs opened 5 months ago

frozenbugs commented 5 months ago

🔨Work Item

IMPORTANT:

Project tracker: https://github.com/orgs/dmlc/projects/2

Description

Depending work items or issues

mfbalin commented 5 months ago

image

mfbalin commented 2 months ago

3960 is related.

mfbalin commented 2 months ago

This is the main reason Graphbolt's DGL examples will be much slower compared to PyG examples. We are eliminating all CPU GPU synchronizations from the code and this will remain the only synchronization when we are done.

mfbalin commented 2 months ago

If this issue is not resolved, we won't be able to showcase GraphBolt's peak performance on DGL examples.

mfbalin commented 2 months ago

@TristonC