Skip to content

Commit 2b1a40f

Browse files
committed
integration test
1 parent ab033dd commit 2b1a40f

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/integration_tests/features.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -346,6 +346,18 @@ def build_features_test_list() -> list[OverrideDefinitions]:
346346
"fsdp+flex_attn+per_op_sac",
347347
ngpu=4,
348348
),
349+
OverrideDefinitions(
350+
[
351+
[
352+
"--parallelism.data_parallel_shard_degree=4",
353+
"--activation_checkpoint.mode='full'",
354+
"--model.flavor=debugmodel_varlen_attn",
355+
]
356+
],
357+
"FSDP+VARLEN_ATTN",
358+
"fsdp+varlen_attn",
359+
ngpu=4,
360+
),
349361
OverrideDefinitions(
350362
[
351363
[

0 commit comments

Comments
 (0)