PyThread_set_key_value does not overwrite in python 2.7.x, so delete …

…first
cupy · Oct 23, 2017 · 8163801 · 8163801
1 parent 1c1a34b
commit 8163801
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cupy/cuda/stream.pyx b/cupy/cuda/stream.pyx
@@ -41,6 +41,7 @@ cpdef _set_current_stream(stream):
     if stream is None:
         stream = Stream.null
     cdef size_t stream_ptr = stream.ptr
+    pythread.PyThread_delete_key_value(_current_stream_key)
     pythread.PyThread_set_key_value(_current_stream_key, <void *>stream_ptr)
     _thread_local.current_stream_ref = weakref.ref(stream)