mirrors / dmlc--dgl

文件历史

2,455 提交 34 分支 42 Git 标签
0.9.x
提交图

2 次代码提交

作者 SHA1 备注 提交日期
nv-dlasalle 1425150459 Fix test naming (#4023) 
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2022-05-23 17:21:06 +08:00
nv-dlasalle 7f9279397a [Feature][DistDGL] Add NCCL support for range based partitions (#3213) 
* 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
 2021-08-19 21:17:55 -07:00