"jenkins.sh" did not exist on "7374ed8cb35e56944bb7465c276f7af5eb3c489d"
  • 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
partition_op.cu 22 KB