Skip to content

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

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

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

Annotations

1 warning

pytester (macOS-14, 3.10, latest, core)

succeeded Dec 17, 2024 in 9m 58s