[Test] Set DP size in the mocked Metal profiler test (#40667)

2026-09-21 19:57:52 -07:00
parent 90cf471723
commit 1d025491f3
1 changed files with 1 additions and 1 deletions
@@ -261,7 +261,7 @@ class TestSchedulerProfilerManagerMPS(unittest.TestCase):
                    torch.mps.profiler, "metal_capture", return_value=capture_ctx
                ),
                mock_patch("torch.distributed.barrier"),
-                get_parallel().override(tp_rank=0, pp_size=1, moe_ep_size=1),
+                get_parallel().override(tp_rank=0, dp_size=1, pp_size=1, moe_ep_size=1),
            ):
                result = mgr._start_profile()
                self.assertTrue(result.success, result.message)