Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "5f8c31835a0871fadaf99fc4f49d56f774890dfa"
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
test_partition.py 24.5 KB