• kylasa's avatar
    [DistDGL][Optimizations]Rehash code to optimize for loop (#5224) · 9ce800d2
    kylasa authored
    * Rehash code to optimize for loop
    
    Reduced number of instructions in for loop, which exchanging edge features. This will reduce the number of times numpy's intersect1d is invoked (saving the runtime and memory overhead needs of numpy).
    
    * Applying lintrunner patch to data_shuffle.py
    9ce800d2
data_shuffle.py 51.3 KB