Skip to content

Commit 1af38e5

Browse files
committed
integration test
1 parent ab033dd commit 1af38e5

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

tests/integration_tests/models.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,25 @@ def build_model_tests_list() -> list[OverrideDefinitions]:
110110
"llama4_pp+fsdp+tp+ep+compile",
111111
ngpu=8,
112112
),
113+
# Integration Test Cases for Llama 3 Varlen
114+
OverrideDefinitions(
115+
[
116+
[
117+
"--model.name llama3",
118+
"--model.flavor debugmodel_varlen",
119+
"--parallelism.pipeline_parallel_degree 2",
120+
"--parallelism.pipeline_parallel_schedule Interleaved1F1B",
121+
"--parallelism.data_parallel_shard_degree 2",
122+
"--parallelism.tensor_parallel_degree 2",
123+
"--parallelism.expert_parallel_degree 4",
124+
"--parallelism.expert_tensor_parallel_degree 1",
125+
"--compile.enable",
126+
],
127+
],
128+
"Llama 3 PP+FSDP+TP+EP+compile+varlen",
129+
"llama3_pp+fsdp+tp+ep+compile+varlen",
130+
ngpu=8,
131+
),
113132
]
114133

115134
return model_tests

0 commit comments

Comments
 (0)