ColossalAI/tests/test_booster
flybird11111 5e16bf7980
[shardformer] fix gathering output when using tensor parallelism (#5431)
* fix

* padding vocab_size when using pipeline parallellism

padding vocab_size when using pipeline parallellism

fix

fix

* fix

* fix

fix

fix

* fix gather output

* fix

* fix

* fix

fix resize embedding

fix resize embedding

* fix resize embedding

fix

* revert

* revert

* revert
2024-03-18 15:55:11 +08:00
..
test_mixed_precision [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
test_plugin [shardformer] fix gathering output when using tensor parallelism (#5431) 2024-03-18 15:55:11 +08:00
test_accelerator.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00