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 |
||
---|---|---|
.. | ||
__init__.py | ||
bnb.py | ||
bnb_config.py | ||
fp8.py | ||
fp8_hook.py | ||
utils.py |