You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_layers/test_3d/checks_3d
zbian e94c79f15b
improved allgather & reducescatter for 3d
2 years ago
..
__init__.py added CI for unit testing (#69) 3 years ago
check_layer_3d.py improved allgather & reducescatter for 3d 2 years ago
common.py updated tp layers 2 years ago