From 27134562e360de3dddbd423bd37b658af6a3f98b Mon Sep 17 00:00:00 2001 From: Jon Bolin Date: Mon, 30 Oct 2023 20:07:18 +0000 Subject: [PATCH] Remove master IP discovery test for MP --- test/pjrt/test_runtime_tpu.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/test/pjrt/test_runtime_tpu.py b/test/pjrt/test_runtime_tpu.py index d7a7918bc9c..174c7313a79 100644 --- a/test/pjrt/test_runtime_tpu.py +++ b/test/pjrt/test_runtime_tpu.py @@ -238,13 +238,6 @@ def test_execute_time_metric(self): f"Expected exectue time of {i} to take more than " f"{expected_time_seconds} seconds, got {v / 1e9} seconds") - @mock.patch('torch_xla._internal.tpu.get_worker_ips') - def test_master_ip_discovery(self, patched_get_worker_ips): - # A basic test to verify the non-SPMD codepath returns the correct IP. Two - # IPs are needed to avoid the short-circuit return of localhost. - patched_get_worker_ips.return_value = ['10.0.0.1', '10.0.0.2'] - self.assertTrue(xr.get_master_ip(), '10.0.0.1') - if __name__ == '__main__': absltest.main()