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/colossalai/quantization
Hongxin Liu 5ddad486ca
[fp8] add fallback and make compile option configurable (#6092)
1 month ago
..
__init__.py
bnb.py [quant] fix bitsandbytes version check (#5882) 5 months ago
bnb_config.py
fp8.py [fp8] add fallback and make compile option configurable (#6092) 1 month ago
fp8_config.py [fp8] add fallback and make compile option configurable (#6092) 1 month ago
fp8_hook.py [fp8] support gemini plugin (#5978) 4 months ago
utils.py [Feature]: support FP8 communication in DDP, FSDP, Gemini (#5928) 4 months ago