torchx - profile scheduler validate call

Differential Revision: D66166216 Pull Request resolved: #972
pytorch · Jan 10, 2025 · ffb6cef · ffb6cef
1 parent 5e44f20
commit ffb6cef
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/torchx/runner/api.py b/torchx/runner/api.py
@@ -433,7 +433,11 @@ def dryrun(
                         " Either a patch was built or no changes to workspace was detected."
                     )
 
-            sched._validate(app, scheduler, resolved_cfg)
+            with log_event(
+                "validate",
+                scheduler,
+            ):
+                sched._validate(app, scheduler, resolved_cfg)
             dryrun_info = sched.submit_dryrun(app, resolved_cfg)
             dryrun_info._scheduler = scheduler
             return dryrun_info