From 9a0dca78759b0c5f74f19f4b78b964b1ac584250 Mon Sep 17 00:00:00 2001 From: jonb377 Date: Mon, 30 Oct 2023 13:13:18 -0700 Subject: [PATCH] Remove master IP discovery test for MP (#5748) --- 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 d7a7918bc9cc..174c7313a796 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()