search

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.55k stars 3.01k forks source link

Parallelize the LineGraph implementation to speed up in multi-core machines #4129

Open bioannidis opened 2 years ago

bioannidis commented 2 years ago

https://github.com/dmlc/dgl/blob/230b886ec55b9927bc806848e30c3a9b33bbcd6e/python/dgl/transforms/module.py#L764

Current LineGraph implementation takes long time in large graphs (15M edges-8M nodes). Could we speed up the implementation of this transform?

jermainewang commented 2 years ago

Could you also provide some contexts for us to understand your using scenarios? For example, is the transformation done once or at each iteration? How long it takes currently for the scale of your graph?

bioannidis commented 2 years ago

It is done once. I started the code yesterday it runs for 25 hours no result yet.