reenable compile

Author: acisseJZhong

tests/integration_tests/models.py

@@ -173,7 +173,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",
                 ],
             ],
             "Llama 4 PP+FSDP+TP+EP+compile",
@@ -206,7 +206,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",

torchtitan/models/llama4/config_registry.py

@@ -87,7 +87,7 @@ def llama4_debugmodel_fp8() -> Trainer.Config:
         activation_checkpoint=ActivationCheckpointConfig(
             mode="selective",
         ),
-        compile=CompileConfig(enable=False, components=["model", "loss"]),
+        compile=CompileConfig(enable=True, components=["model", "loss"]),
         model_converters=ModelConvertersContainer.Config(
             converters=[
                 Float8GroupedMMConverter.Config(fqns=["experts"]),