File tree Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments