Skip to content

Commit 33e8791

Browse files
authored
Partially disable tpu-info CLI tests (#9463)
1 parent 5d05f0a commit 33e8791

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

test/tpu/tpu_info/test_cli.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,9 @@ def test_single_process_e2e(self, extra_env):
6868
self.assertEqual(u.duty_cycle_pct, 0.0)
6969
one_gb = 1 << 30
7070
self.assertLess(u.memory_usage, one_gb)
71-
# TODO: check output
72-
cli.print_chip_info()
71+
# TODO(https://github.com/pytorch/xla/issues/9462): Uncomment after
72+
# libtpu is fixed for python 3.12
73+
# cli.print_chip_info()
7374

7475
@contextlib.contextmanager
7576
def _torch_xla_spawn(self):
@@ -104,8 +105,9 @@ def test_multiprocessing_e2e(self):
104105
self.assertEqual(u.duty_cycle_pct, 0.0)
105106
one_gb = 1 << 30
106107
self.assertLess(u.memory_usage, one_gb)
107-
# TODO: check output
108-
cli.print_chip_info()
108+
# TODO(https://github.com/pytorch/xla/issues/9462): Uncomment after
109+
# libtpu is fixed for python 3.12
110+
# cli.print_chip_info()
109111

110112

111113
if __name__ == "__main__":

0 commit comments

Comments
 (0)