diff --git a/test_runner.py b/test_runner.py index 6a7b6b1a53..b0eb04c91b 100755 --- a/test_runner.py +++ b/test_runner.py @@ -305,6 +305,18 @@ def build_test_list(): "FSDP2 with float8 all-gather and precomputed dynamic scales", "fsdp2_float8_all_gather_precompute_dynamic_scales", ), + OverrideDefinitions( + [ + [ + "--training.enable_float8_linear", + "--training.enable_fsdp_float8_all_gather", + "--training.precompute_float8_dynamic_scale_for_fsdp", + "--training.compile", + ] + ], + "FSDP2 with float8 all-gather and precomputed dynamic scales", + "fsdp2_float8_all_gather_precompute_dynamic_scales_compile", + ), OverrideDefinitions( [ [