• Muhammed Fatih BALIN's avatar
    [Feature] (La)yer-Neigh(bor) sampling implementation (#4668) · bf264d00
    Muhammed Fatih BALIN authored
    
    
    * adding LABOR sampling
    
    * add ladies and pladies samplers
    
    * fix compile error after rebase
    
    * add reference for ladies sampler
    
    * Improve ladies implementation.
    
    * weighted labor sampling initial implementation draft
    fix indentation and small bug in ladies script
    
    * importance_sampling currently doesn't work with weights
    
    * fix weighted importance sampling
    
    * move labor example into its own folder
    
    * lint fixes
    
    * Improve documentation
    
    * remove examples from the main PR
    
    * fix linting by not using c++17 features
    
    * fix documentation of labor_sampler.py
    
    * update documentation for labor.py
    
    * reformat the labor.py file with black
    
    * fix linting errors
    
    * replace exception use with if
    
    * fix typo in error comment
    
    * fixing win64 build for ci
    
    * fixing weighted implementation, works now.
    
    * fix bug in the weighted case and importance_sampling==0
    
    * address part of the reviews
    
    * remove unused code paths from cuda
    
    * remove unused code path from cpu side
    
    * remove extra features of labor making use of random seed.
    
    * fix exclude_edges bug
    
    * remove pcg and seed logic from cpu implementation, seed logic should still work for cuda.
    
    * minor style change
    
    * refactor CPU implementation, take out the importance_sampling probability computation into a function.
    
    * improve CUDAWorkspaceAllocator
    
    * refactor importance_sampling part out to a function
    
    * minor optimization
    
    * fix linting issue
    
    * Revert "remove pcg and seed logic from cpu implementation, seed logic should still work for cuda."
    
    This reverts commit c250e07ac6d7e13f57e79e8a2c2f098d777378c2.
    
    * Revert "remove extra features of labor making use of random seed."
    
    This reverts commit 7f99034353080308f4783f27d9a08bea343fb796.
    
    * fix the documentation
    
    * disable NIDs
    
    * improve the documentation in the code
    
    * use the stream argument in pcg32 instead of skipping ahead t times, can discard the use of hashmap now since it is faster this way.
    
    * fix linting issue
    
    * address another round of reviews
    
    * further optimize CPU LABOR sampling implementation
    
    * fix linting error
    
    * update the comment
    
    * reformat
    
    * rename and rephrase comment
    
    * fix formatting according to new linting specs
    
    * fix compile error due to renaming, fix linting.
    
    * lint
    
    * rename DGLHeteroGraph to DGLGraph to match master
    
    * replace other occurrences of DGLHeteroGraph to DGLGraph
    Co-authored-by: default avatarMuhammed Fatih BALIN <m.f.balin@gmail.com>
    Co-authored-by: default avatarKaan Sancak <kaansnck@gmail.com>
    Co-authored-by: default avatarQuan Gan <coin2028@hotmail.com>
    bf264d00
labor_pick.h 11.2 KB