Fix test.

pytorch · Nov 1, 2024 · ca03671 · ca03671
1 parent 7c7ad4e
commit ca03671
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/test_operations.py b/test/test_operations.py
@@ -2916,8 +2916,7 @@ def test_dlpack_xla_to_pytorch_cuda(self):
   @onlyIfPJRTDeviceIsCUDA
   def test_dlpack_xla_to_pytorch_cuda_protocol_conversion(self):
     xla_t1 = torch.arange(5).to(xm.xla_device())
-    caps_t1 = torch.utils.dlpack.to_dlpack(xla_t1)
-    cuda_t1 = torch.utils.dlpack.from_dlpack(caps_t1)
+    cuda_t1 = torch.utils.dlpack.from_dlpack(xla_t1)
     self.assertEqual(cuda_t1.device.type, 'cuda')
     self.assertEqual(cuda_t1.device.index, xla_t1.device.index)
     cuda_t1[0] = cuda_t1[0] + 20