There was an error while loading. Please reload this page.
1 parent 1f0e972 commit 595ebcfCopy full SHA for 595ebcf
test/pjrt/test_runtime_tpu.py
@@ -238,13 +238,6 @@ def test_execute_time_metric(self):
238
f"Expected exectue time of {i} to take more than "
239
f"{expected_time_seconds} seconds, got {v / 1e9} seconds")
240
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
-
248
249
if __name__ == '__main__':
250
absltest.main()
0 commit comments