Skip to content

[litgpt benchmark] enable force_recompute_fp8_weight_in_bwd when torchao.float8 is used with FSDP2#1528

Open
crcrpar wants to merge 1 commit intomainfrom crpa/ao-recompute_fp8_weight_in_bwd

Commits

Commits on Dec 17, 2024