Fix broken test (#505)

Sam Lurye · facebook-github-bot · commit 324e6f7a8e8b · 2025-07-11T11:47:17.000-07:00
Summary: Pull Request resolved: #505 Fix broken test for T230654436. `DebugManager` should only be spawned when the proc mesh is first created, similar to `RDMAManager`. If `_mock_shape` isn't None, it means we're creating a proc mesh by slicing an existing one, so DebugManager will already exist. Reviewed By: colin2328, dcci Differential Revision: D78171244 fbshipit-source-id: 0f44ab40b3e530c9266d69f68a0b38669fa2d977
diff --git a/python/monarch/_src/actor/proc_mesh.py b/python/monarch/_src/actor/proc_mesh.py
@@ -103,7 +103,7 @@ def __init__(
         if _mock_shape is None and HAS_TENSOR_ENGINE:
             # type: ignore[21]
             self._rdma_manager = self._spawn_blocking("rdma_manager", RDMAManager)
-        if not _is_initializing_debugger:
+        if not _is_initializing_debugger and _mock_shape is None:
             self._debug_manager = self._spawn_blocking(
                 _DEBUG_MANAGER_ACTOR_NAME, DebugManager, debug_client()
             )

Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ def __init__(`
`103`	`103`	`if _mock_shape is None and HAS_TENSOR_ENGINE:`
`104`	`104`	`# type: ignore[21]`
`105`	`105`	`self._rdma_manager = self._spawn_blocking("rdma_manager", RDMAManager)`
`106`		`- if not _is_initializing_debugger:`
	`106`	`+ if not _is_initializing_debugger and _mock_shape is None:`
`107`	`107`	`self._debug_manager = self._spawn_blocking(`
`108`	`108`	`_DEBUG_MANAGER_ACTOR_NAME, DebugManager, debug_client()`
`109`	`109`	`)`