From 1d025491f3448f52d6e1df47958e681151ce3167 Mon Sep 17 00:00:00 2001 From: Cheng Wan <54331508+ch-wan@users.noreply.github.com> Date: Mon, 21 Sep 2026 19:57:52 -0700 Subject: [PATCH] [Test] Set DP size in the mocked Metal profiler test (#40667) --- .../registered/unit/hardware_backend/mlx/test_metal_profiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/registered/unit/hardware_backend/mlx/test_metal_profiler.py b/test/registered/unit/hardware_backend/mlx/test_metal_profiler.py index e21b7a64e..97c8bb412 100644 --- a/test/registered/unit/hardware_backend/mlx/test_metal_profiler.py +++ b/test/registered/unit/hardware_backend/mlx/test_metal_profiler.py @@ -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)