Skip to content

Commit d76b77f

Browse files
authored
remove CI for FSDP2 + fp8 all-gather (#470)
per discussion from #469 (comment) we are planning BC breaking changes in float8_experimental. remove CI for FSDP2 + fp8 all-gather for now. When public APIs are finalized, we can discuss bringing it back
1 parent b502cdc commit d76b77f

File tree

1 file changed

+0
-44
lines changed

1 file changed

+0
-44
lines changed

test_runner.py

Lines changed: 0 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -273,50 +273,6 @@ def build_test_list():
273273
"fsdp2_mem_tracker",
274274
ngpu=4,
275275
),
276-
OverrideDefinitions(
277-
[
278-
[
279-
"--training.enable_float8_linear",
280-
]
281-
],
282-
"FSDP2 with original dtype",
283-
"float8_fsdp2_orig_all_gather",
284-
ngpu=4,
285-
),
286-
OverrideDefinitions(
287-
[
288-
[
289-
"--training.enable_float8_linear",
290-
"--training.enable_fsdp_float8_all_gather",
291-
]
292-
],
293-
"FSDP2 with float8 all-gather",
294-
"fsdp2_float8_all_gather",
295-
ngpu=4,
296-
),
297-
OverrideDefinitions(
298-
[
299-
[
300-
"--training.enable_float8_linear",
301-
"--training.enable_fsdp_float8_all_gather",
302-
"--training.precompute_float8_dynamic_scale_for_fsdp",
303-
]
304-
],
305-
"FSDP2 with float8 all-gather and precomputed dynamic scales",
306-
"fsdp2_float8_all_gather_precompute_dynamic_scales",
307-
),
308-
OverrideDefinitions(
309-
[
310-
[
311-
"--training.enable_float8_linear",
312-
"--training.enable_fsdp_float8_all_gather",
313-
"--training.precompute_float8_dynamic_scale_for_fsdp",
314-
"--training.compile",
315-
]
316-
],
317-
"FSDP2 with float8 all-gather and precomputed dynamic scales",
318-
"fsdp2_float8_all_gather_precompute_dynamic_scales_compile",
319-
),
320276
OverrideDefinitions(
321277
[
322278
[

0 commit comments

Comments
 (0)