Skip to content

Commit 1fc53ec

Browse files
authored
Merge pull request #7236 from gmarkall/issue-7180
CUDA: Skip managed alloc tests on ARM
2 parents 01dc3b4 + db5a12f commit 1fc53ec

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

‎numba/cuda/api.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -146,8 +146,8 @@ def managed_array(shape, dtype=np.float_, strides=None, order='C', stream=0,
146146
Allocate a np.ndarray with a buffer that is managed.
147147
Similar to np.empty().
148148
149-
Managed memory is supported on Linux, and is considered experimental on
150-
Windows.
149+
Managed memory is supported on Linux / x86 and PowerPC, and is considered
150+
experimental on Windows and Linux / AArch64.
151151
152152
:param attach_global: A flag indicating whether to attach globally. Global
153153
attachment implies that the memory is accessible from

‎numba/cuda/tests/cudadrv/test_managed_alloc.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,13 @@
33
from numba.cuda.cudadrv.driver import device_memset, driver
44
from numba import cuda
55
from numba.cuda.testing import unittest, ContextResettingTestCase
6-
from numba.cuda.testing import skip_on_cudasim
6+
from numba.cuda.testing import skip_on_cudasim, skip_on_arm
77
from numba.tests.support import linux_only
88

99

1010
@skip_on_cudasim('CUDA Driver API unsupported in the simulator')
1111
@linux_only
12+
@skip_on_arm('Managed Alloc support is experimental/untested on ARM')
1213
class TestManagedAlloc(ContextResettingTestCase):
1314

1415
def get_total_gpu_memory(self):

0 commit comments

Comments
 (0)