xgqdut2016
|
a3929c25f8
|
Add send and recv operators based on NCCL (#182)
* baseline sendrecv, bug
* success sendrecv
* get rank from comm
* set output shape
* successful:set output shape equal to input shape
* shape as attribute
* success:shape as attribute
* success send recv, output 0
* add onnx test
* split send and recv
* success split send and recv
* test-onnx bug
* success test-onnx
* modified onnx.py
* solve review
|
2023-12-14 16:38:03 +08:00 |