Use pointers

Author: crusaderky

tests/test_cupy.py

@@ -12,11 +12,6 @@
         lambda: Stream(non_blocking=True),
         lambda: Stream(null=True),
         lambda: Stream(ptds=True),
-        pytest.param(
-            lambda: 123,
-            id="dlpack stream",
-            marks=pytest.mark.skip(reason="segmentation fault reported (#326)")
-        ),
     ],
 )
 def test_to_device_with_stream(make_stream):
@@ -32,3 +27,19 @@ def test_to_device_with_stream(make_stream):
         # device context.
         b = to_device(a, dev, stream=stream)
         assert device(b) == dev
+
+
+def test_to_device_with_dlpack_stream():
+    devices = xp.__array_namespace_info__().devices()
+
+    a = xp.asarray([1, 2, 3])
+    for dev in devices:
+        # Streams are device-specific and must be created within
+        # the context of the device...
+        with dev:
+            s1 = Stream()
+
+        # ... however, to_device() does not need to be inside the
+        # device context.
+        b = to_device(a, dev, stream=s1.ptr)
+        assert device(b) == dev