fix imports

AMYPAD · Feb 18, 2024 · 224eba0 · 224eba0
1 parent 71e7531
commit 224eba0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/tests/test_cpython.py b/tests/test_cpython.py
@@ -111,17 +111,18 @@ def test_asarray():
 
 
 def test_cuda_array_interface():
+    dev_sync = importorskip("cuvec.dev_sync")
     cupy = importorskip("cupy")
     v = cu.asarray(np.random.random(shape))
     assert hasattr(v, '__cuda_array_interface__')
 
     c = cupy.asarray(v)
     assert (c == v).all()
     c[0, 0, 0] = 1
-    cu.dev_sync()
+    dev_sync()
     assert c[0, 0, 0] == v[0, 0, 0]
     c[0, 0, 0] = 0
-    cu.dev_sync()
+    dev_sync()
     assert c[0, 0, 0] == v[0, 0, 0]
 
     ndarr = v + 1

diff --git a/tests/test_pybind11.py b/tests/test_pybind11.py
@@ -3,8 +3,6 @@
 import numpy as np
 from pytest import importorskip, mark, raises
 
-from cuvec import dev_sync
-
 cu = importorskip("cuvec.pybind11")
 shape = 127, 344, 344
 
@@ -119,6 +117,7 @@ def test_asarray():
 
 
 def test_cuda_array_interface():
+    dev_sync = importorskip("cuvec.dev_sync")
     cupy = importorskip("cupy")
     v = cu.asarray(np.random.random(shape))
     assert hasattr(v, '__cuda_array_interface__')

diff --git a/tests/test_swig.py b/tests/test_swig.py
@@ -3,8 +3,6 @@
 import numpy as np
 from pytest import importorskip, mark, raises
 
-from cuvec import dev_sync
-
 cu = importorskip("cuvec.swig")
 shape = 127, 344, 344
 
@@ -119,6 +117,7 @@ def test_asarray():
 
 
 def test_cuda_array_interface():
+    dev_sync = importorskip("cuvec.dev_sync")
     cupy = importorskip("cupy")
     v = cu.asarray(np.random.random(shape))
     assert hasattr(v, '__cuda_array_interface__')