Guangyao Zhang f20b066c59 [fp8] Disable all_gather intranode. Disable Redundant all_gather fp8 (#6059) ... * 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		2024-09-14 10:40:01 +08:00
..
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
chunk.py	[fp8] Disable all_gather intranode. Disable Redundant all_gather fp8 (#6059)	2024-09-14 10:40:01 +08:00
manager.py	[Feature]: support FP8 communication in DDP, FSDP, Gemini (#5928)	2024-08-08 15:55:01 +08:00
search_utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
utils.py	[Gemini] Use async stream to prefetch and h2d data moving (#5781)	2024-06-12 15:48:52 +08:00