From f50e471e05f170e827d6bf8030b0924d29de1deb Mon Sep 17 00:00:00 2001 From: Mason Chang Date: Thu, 7 Mar 2024 14:52:10 +0000 Subject: [PATCH] Clarify comments --- test/dynamo/test_dynamo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/dynamo/test_dynamo.py b/test/dynamo/test_dynamo.py index fcecb09ceb24..c5bedeaa41b3 100644 --- a/test/dynamo/test_dynamo.py +++ b/test/dynamo/test_dynamo.py @@ -103,9 +103,9 @@ def test_simple_model(self): self.assertNotIn('XLAData: None', torch_xla._XLAC._get_xla_tensor_debug_info(xla_y3)) - # Tests that the dynamo bridge automatically moves tensors to XLA device + # Tests that the dynamo bridge automatically moves tensors to XLA device, + # then back to the original device. def test_simple_model_automoves_tensors(self): - xla_device = xm.xla_device() x = torch.tensor(100.0) y = torch.tensor(200.0) eager_result = self.fn_simple(x, y)