diff --git a/tests/integration_tests/models.py b/tests/integration_tests/models.py index 83a4c9ad75..c033ccca0c 100755 --- a/tests/integration_tests/models.py +++ b/tests/integration_tests/models.py @@ -196,8 +196,6 @@ def build_model_tests_list() -> list[OverrideDefinitions]: ngpu=8, ), # Integration Test Cases for gpt-oss - # TODO: re-enable compile after fixing - # https://github.com/pytorch/torchtitan/issues/2776 OverrideDefinitions( [ [ @@ -206,7 +204,7 @@ def build_model_tests_list() -> list[OverrideDefinitions]: "--parallelism.tensor_parallel_degree 2", "--parallelism.expert_parallel_degree 4", "--parallelism.expert_tensor_parallel_degree 1", - # "--compile.enable", + "--compile.enable", ], ], "Gpt-oss FSDP+TP+EP+compile",