zbian e94c79f15b improved allgather & reducescatter for 3d

2023-01-03 17:46:08 +08:00

..

__init__.py

added CI for unit testing (#69)

2021-12-16 10:32:08 +08:00

check_layer_3d.py

improved allgather & reducescatter for 3d

2023-01-03 17:46:08 +08:00

common.py

updated tp layers

2022-11-02 12:19:38 +08:00