diff --git a/test_runner.py b/test_runner.py index b0eb04c91b..67117bfed6 100755 --- a/test_runner.py +++ b/test_runner.py @@ -273,50 +273,6 @@ def build_test_list(): "fsdp2_mem_tracker", ngpu=4, ), - OverrideDefinitions( - [ - [ - "--training.enable_float8_linear", - ] - ], - "FSDP2 with original dtype", - "float8_fsdp2_orig_all_gather", - ngpu=4, - ), - OverrideDefinitions( - [ - [ - "--training.enable_float8_linear", - "--training.enable_fsdp_float8_all_gather", - ] - ], - "FSDP2 with float8 all-gather", - "fsdp2_float8_all_gather", - ngpu=4, - ), - OverrideDefinitions( - [ - [ - "--training.enable_float8_linear", - "--training.enable_fsdp_float8_all_gather", - "--training.precompute_float8_dynamic_scale_for_fsdp", - ] - ], - "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( [ [