Skip to content

(fsdp2 dev. perf) Add training.fp8_linear: torchao Float8Linear with float8 all-gather - #923

Draft
yushengsu-thu wants to merge 9 commits into
sgl-project:mainfrom
yushengsu-thu:fsdp2/fp8-linear
Draft

(fsdp2 dev. perf) Add training.fp8_linear: torchao Float8Linear with float8 all-gather#923
yushengsu-thu wants to merge 9 commits into
sgl-project:mainfrom
yushengsu-thu:fsdp2/fp8-linear

Add training.fp8_linear for the FSDP2 backend

a063ccf
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs