• ChongyuNVIDIA's avatar
    Add the permutation related support as the extension for asp lib. (#1194) · 89edb819
    ChongyuNVIDIA authored
    * Add the permutation related support as the extension for asp lib.
    
    * [Fix] Track the permutation sequence for progressive channel swap strategy.
    
    * Fix the corner case that one layer is not sparse, but need to apply permutation due to its siblings.
    
    * Fix the deprecated functions in ASP unit tests.
    
    * Fix the sparsity info typo in ASP lib.
    
    * [Enhancement] Set the identical random seed for all GPUs to make sure the same results generated in permutation search.
    
    * Update the README.md with identical random seed setting and NeurIPS info.
    
    * Integrate the Pybind11 enhancement of permutation search into ASP lib.
    89edb819
setup.py 24 KB