addressing old comments

Author: brandon-b-miller

numba_cuda/numba/cuda/cudadrv/driver.py

@@ -3536,7 +3536,8 @@ def _stream_handle(stream):
             "Expected a Stream object or 0, got %s" % type(stream).__name__
         )
     elif hasattr(stream, "__cuda_stream__"):
-        _, ptr = stream.__cuda_stream__()
+        ver, ptr = stream.__cuda_stream__()
+        assert ver == 0
         if USE_NV_BINDING and isinstance(ptr, binding.CUstream):
             return get_cuda_native_handle(ptr)
         else:

numba_cuda/numba/cuda/tests/cudadrv/test_cuda_driver.py

@@ -213,6 +213,7 @@ def kernel(a):
         stream.sync()
 
         kernel[1, 100, stream](ary)
+        stream.sync()
 
         result = ary.copy_to_host(stream=stream)
         for i, v in enumerate(result):