InfiniTensor/test
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
..
core Add ReduceSum op and kernel (#160) 2023-11-24 09:29:58 +08:00
cuda impl distributed launch with NCCL (#106) 2023-09-05 09:47:35 +08:00
kernels Add send and recv operators based on NCCL (#182) 2023-12-14 16:38:03 +08:00
nnet test: 支持编译 einnet 单元测试,但不是所有测试都能通过 (#174) 2023-11-03 13:21:49 +08:00
operators Add send and recv operators based on NCCL (#182) 2023-12-14 16:38:03 +08:00
script build: 实现格式化 git added c/c++ 源码的脚本 (#98) 2023-07-21 12:29:50 +08:00