InfiniTensor/test/kernels
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
..
bang Pooling ceil mode (#155) 2023-10-09 20:51:39 +08:00
cuda Add send and recv operators based on NCCL (#182) 2023-12-14 16:38:03 +08:00
intelcpu memory_allocator (#103) 2023-08-13 13:39:35 +08:00
kunlun Xpu task4 support: add softmax (#172) 2023-10-30 16:01:05 +08:00
nativecpu add transpose, concat and split for native cpu (#158) 2023-10-12 10:14:28 +08:00