File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed
torchtitan/experiments/torchcomms Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -30,21 +30,23 @@ def build_torchcomms_test_list() -> list[OverrideDefinitions]:
3030 [
3131 "--parallelism.tensor_parallel_degree 2" ,
3232 "--parallelism.pipeline_parallel_degree 2" ,
33+ "--compile.enable" ,
3334 ],
3435 ],
35- "FSDP+TP+PP" ,
36- "3d_dp+tp+pp" ,
36+ "FSDP+TP+PP+compile " ,
37+ "3d_dp+tp+pp+compile " ,
3738 ngpu = 8 ,
3839 ),
3940 OverrideDefinitions (
4041 [
4142 [
4243 "--parallelism.context_parallel_degree 2" ,
4344 "--parallelism.pipeline_parallel_degree 2" ,
45+ "--compile.enable" ,
4446 ],
4547 ],
46- "DP+CP+PP" ,
47- "3d_dp+cp+pp" ,
48+ "DP+CP+PP+compile " ,
49+ "3d_dp+cp+pp+compile " ,
4850 ngpu = 8 ,
4951 ),
5052 # TODO: Enable async TP test once fixes are available for running on CI nodes.
You can’t perform that action at this time.
0 commit comments