Skip to content

Commit

Permalink
Remove master IP discovery test for MP (#5748)
Browse files Browse the repository at this point in the history
  • Loading branch information
jonb377 committed Oct 31, 2023
1 parent 25dc184 commit 9a0dca7
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions test/pjrt/test_runtime_tpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

0 comments on commit 9a0dca7

Please sign in to comment.