mirror of https://github.com/hpcaitech/ColossalAI
f20b066c59
* all_gather only internode, fix pytest * fix cuda arch <89 compile pytest error * fix pytest failure * disable all_gather_into_tensor_flat_fp8 * fix fp8 format * fix pytest * fix conversations * fix chunk tuple to list |
||
---|---|---|
.. | ||
test_all_to_all_single.py | ||
test_fp8_all_to_all.py | ||
test_fp8_all_to_all_single.py | ||
test_fp8_allgather.py | ||
test_fp8_allreduce.py | ||
test_fp8_cast.py | ||
test_fp8_ddp_comm_hook.py | ||
test_fp8_fsdp_comm_hook.py | ||
test_fp8_hook.py | ||
test_fp8_linear.py | ||
test_fp8_reduce_scatter.py |