Skip to content

Commit e96bb4a

Browse files
committed
bug fix
1 parent 227d9c1 commit e96bb4a

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

cuda_core/cuda/core/experimental/_stream.pyx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -262,6 +262,7 @@ cdef class Stream:
262262
# on the stream. Event flags such as disabling timing, nonblocking,
263263
# and CU_EVENT_RECORD_EXTERNAL, can be set in EventOptions.
264264
if event is None:
265+
self._get_device_and_context()
265266
event = Event._init(self._device_id, self._ctx_handle, options)
266267
err, = driver.cuEventRecord(event.handle, self._handle)
267268
raise_if_driver_error(err)

0 commit comments

Comments
 (0)