Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "a3faf3f260101c2bc09112ebb514d1e0d0220e1a"
Switch branch/tag
Find file BlameHistoryPermalink
  • Rhett Ying's avatar
    [Dist] Reduce peak memory in DistDGL (#4687) · b1309217
    Rhett Ying authored 
    * [Dist] Reduce peak memory in DistDGL: avoid validation, release memory once loaded

* remove orig_id from ndata/edata for partition_graph()

* delete orig_id from ndata/edata in dist part pipeline

* reduce dtype size and format before saving graphs

* fix lint

* ETYPE requires to be int32/64 for CSRSortByTag

* fix test failure

* refine
    b1309217
utils.py 17.5 KB