DLPack: fix test using PyTorch API. (#8348)

pytorch · Dec 5, 2024 · 4c99d21 · 4c99d21
1 parent a1a1145
commit 4c99d21
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/test_operations.py b/test/test_operations.py
@@ -2993,8 +2993,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