Move location of test

Author: mdboom

cuda_bindings/benchmarks/test_cupy.py

@@ -12,7 +12,6 @@
 except ImportError:
     skip_tests = True
 
-from cuda.core.experimental.utils import StridedMemoryView
 from kernels import kernel_string
 
 
@@ -198,11 +197,3 @@ def test_launch_latency_small_kernel_16_args(benchmark):
     with stream:
         benchmark(launch, kernel, args)
         stream.synchronize()
-
-
-# Ensure that memory views dellocate their reference to dlpack/cupy tensors
-@pytest.mark.skipif(skip_tests, reason="cupy is not installed")
-def test_strided_memory_view_leak(benchmark):
-    for idx in range(1000):
-        arr = cupy.zeros((1024, 1024, 1024), dtype=cupy.uint8)
-        view = StridedMemoryView(arr, stream_ptr=-1)

cuda_core/tests/test_memory.py

@@ -5,14 +5,18 @@
     from cuda.bindings import driver
 except ImportError:
     from cuda import cuda as driver
-
+try:
+    import cupy
+except ImportError:
+    cupy = None
 import ctypes
 import platform
 
 import pytest
 from cuda.core.experimental import Buffer, Device, DeviceMemoryResource, MemoryResource
 from cuda.core.experimental._memory import DLDeviceType, IPCBufferDescriptor
 from cuda.core.experimental._utils.cuda_utils import handle_return
+from cuda.core.experimental.utils import StridedMemoryView
 
 POOL_SIZE = 2097152  # 2MB size
 
@@ -437,3 +441,11 @@ def test_mempool_attributes_ownership(mempool_device):
     with pytest.raises(RuntimeError, match="DeviceMemoryResource is expired"):
         _ = attributes.used_mem_high
     mr._mempool_handle = old_handle
+
+
+# Ensure that memory views dellocate their reference to dlpack/cupy tensors
+@pytest.mark.skipif(cupy is None, reason="cupy is not installed")
+def test_strided_memory_view_leak(benchmark):
+    for idx in range(1000):
+        arr = cupy.zeros((1024, 1024, 1024), dtype=cupy.uint8)
+        StridedMemoryView(arr, stream_ptr=-1)