mirror of https://github.com/hpcaitech/ColossalAI
53cb9606bd
* add llama shardformer fp8 * Llama Shardformer Parity * fix typo * fix all reduce * fix pytest failure * fix reduce op and move function to fp8.py * fix typo |
||
---|---|---|
.. | ||
test_fp8_all_to_all.py | ||
test_fp8_all_to_all_single.py | ||
test_fp8_allgather_flat.py | ||
test_fp8_allreduce.py | ||
test_fp8_gather.py | ||
test_fp8_reduce_scatter.py |