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 (ubuntu-22.04, 3.10, latest, ops)

succeeded Dec 17, 2024 in 30m 15s