• nv-dlasalle's avatar
    [Feature][DistDGL] Add NCCL support for range based partitions (#3213) · 7f927939
    nv-dlasalle authored
    * Implement range based NDArrayPartition
    
    * Finish implement range based partition support
    
    * Add unit test
    
    * Fix whitepace
    
    * Add Kernel suffix
    
    * Fix argument passing
    
    * Add doxygen docs and improve variable naming
    
    * Add unit test
    
    * Add function for converting a partition book
    
    * Add example to partition_op docs
    
    * Fix dtype conversion for mxnet and tensorflow
    7f927939
test_partition.cc 6.57 KB