From 57a183fd7ab83fe069fd8e930176845b615b4e16 Mon Sep 17 00:00:00 2001 From: Laith Sakka Date: Thu, 5 Dec 2024 14:11:37 -0800 Subject: [PATCH] update test_replace_xla_tensor --- .torch_pin | 1 + test/test_operations.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) create mode 100644 .torch_pin diff --git a/.torch_pin b/.torch_pin new file mode 100644 index 00000000000..2727168a404 --- /dev/null +++ b/.torch_pin @@ -0,0 +1 @@ +#142118 diff --git a/test/test_operations.py b/test/test_operations.py index cc3a73c4580..cb453419431 100644 --- a/test/test_operations.py +++ b/test/test_operations.py @@ -1099,7 +1099,7 @@ def test_replace_xla_tensor(self): t1 = torch.zeros(50, device=xm.xla_device()) t1 += 1 xm.mark_step() - self.assertEqual(met.counter_value('DestroyXlaTensor'), 3) + self.assertEqual(met.counter_value('DestroyXlaTensor'), 1) t2 = torch.zeros(10, device=xm.xla_device()) self.assertEqual(met.counter_value('DestroyXlaTensor'), 4)