• kylasa's avatar
    [DistDGL] Memory optimization to reduce memory footprint of the Dist Graph... · 5ea04713
    kylasa authored
    [DistDGL] Memory optimization to reduce memory footprint of the Dist Graph partitioning pipeline. (#5130)
    
    * Wrap np.argsort() in a function. This
    
    Use a python wrapper for the np.argsort() function for better usage of systems memory.
    
    * lintrunner patch.
    
    * lintrunner patch.
    
    * Changes to address code review comments.
    5ea04713
data_shuffle.py 52 KB