[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
+1