Skip to content

Commit 595ebcf

Browse files
authored
Remove master IP discovery test for MP (#5748)
1 parent 1f0e972 commit 595ebcf

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

test/pjrt/test_runtime_tpu.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -238,13 +238,6 @@ def test_execute_time_metric(self):
238238
f"Expected exectue time of {i} to take more than "
239239
f"{expected_time_seconds} seconds, got {v / 1e9} seconds")
240240

241-
@mock.patch('torch_xla._internal.tpu.get_worker_ips')
242-
def test_master_ip_discovery(self, patched_get_worker_ips):
243-
# A basic test to verify the non-SPMD codepath returns the correct IP. Two
244-
# IPs are needed to avoid the short-circuit return of localhost.
245-
patched_get_worker_ips.return_value = ['10.0.0.1', '10.0.0.2']
246-
self.assertTrue(xr.get_master_ip(), '10.0.0.1')
247-
248241

249242
if __name__ == '__main__':
250243
absltest.main()

0 commit comments

Comments
 (0)