Skip to content

Commit 833485c

Browse files
authored
Take an &Event in Stream::wait_event (#137)
1 parent 1b56eb0 commit 833485c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

crates/cust/src/stream.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,11 +237,11 @@ impl Stream {
237237
/// event.record(&stream_0)?;
238238
///
239239
/// // wait until the work on stream_0 is finished before continuing stream_1
240-
/// stream_1.wait_event(event, StreamWaitEventFlags::DEFAULT)?;
240+
/// stream_1.wait_event(&event, StreamWaitEventFlags::DEFAULT)?;
241241
/// # Ok(())
242242
/// }
243243
/// ```
244-
pub fn wait_event(&self, event: Event, flags: StreamWaitEventFlags) -> CudaResult<()> {
244+
pub fn wait_event(&self, event: &Event, flags: StreamWaitEventFlags) -> CudaResult<()> {
245245
unsafe { cuda::cuStreamWaitEvent(self.inner, event.as_inner(), flags.bits()).to_result() }
246246
}
247247

0 commit comments

Comments
 (0)