frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Temporary disable TestNumbaIntegration.test_from_cuda_array_interface* (#54430)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54430

see https://github.com/pytorch/pytorch/issues/54429

Test Plan: Imported from OSS

Reviewed By: mruberry

Differential Revision: D27232636

Pulled By: pbelevich

fbshipit-source-id: 15fb69828a23cb6f3c173a7863bd55bf4973f408
This commit is contained in:
Pavel Belevich
2021-03-22 09:15:39 -07:00
committed by Facebook GitHub Bot
parent ef472d5b31
commit b6bbb41fd8
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_numba_integration.py
View File

@ -229,6 +229,7 @@ class TestNumbaIntegration(common.TestCase):
numba.cuda.as_cuda_array(cudat), numba.cuda.devicearray.DeviceNDArray
)
@unittest.skip("Test is temporary disabled, see https://github.com/pytorch/pytorch/issues/54418")
@unittest.skipIf(not TEST_NUMPY, "No numpy")
@unittest.skipIf(not TEST_CUDA, "No cuda")
@unittest.skipIf(not TEST_NUMBA_CUDA, "No numba.cuda")
@ -314,6 +315,7 @@ class TestNumbaIntegration(common.TestCase):
torch_ary = torch.as_tensor(numba_ary, device="cuda")
self.assertTrue(torch_ary.is_contiguous())
@unittest.skip("Test is temporary disabled, see https://github.com/pytorch/pytorch/issues/54418")
@unittest.skipIf(not TEST_NUMPY, "No numpy")
@unittest.skipIf(not TEST_CUDA, "No cuda")
@unittest.skipIf(not TEST_NUMBA_CUDA, "No numba.cuda")
@ -325,6 +327,7 @@ class TestNumbaIntegration(common.TestCase):
del numba_ary
self.assertEqual(torch_ary.cpu().data.numpy(), numpy.arange(6)) # `torch_ary` is still alive
@unittest.skip("Test is temporary disabled, see https://github.com/pytorch/pytorch/issues/54418")
@unittest.skipIf(not TEST_NUMPY, "No numpy")
@unittest.skipIf(not TEST_CUDA, "No cuda")
@unittest.skipIf(not TEST_NUMBA_CUDA, "No numba.cuda")