Skip to content

Commit 7929410

Browse files
authored
[TorchComms] add compile in ci test (#1930)
1 parent b01308a commit 7929410

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

torchtitan/experiments/torchcomms/integration_tests.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff 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.

0 commit comments

Comments
 (0)