mircuda-sys 0.1.0

Private native CUDA integration for mircuda
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

use cudarc::driver::{CudaEvent, sys};

use super::driver::{Context, Stream};
use crate::Result;

#[derive(Debug)]
pub struct Event {
    pub(super) inner: CudaEvent,
}

impl Context {
    pub fn create_event(&self, timing: bool) -> Result<Event> {
        let flags = if timing {
            sys::CUevent_flags::CU_EVENT_DEFAULT
        } else {
            sys::CUevent_flags::CU_EVENT_DISABLE_TIMING
        };
        Ok(Event {
            inner: self.inner.new_event(Some(flags))?,
        })
    }
}

impl Event {
    pub fn record(&self, stream: &Stream) -> Result<()> {
        Ok(self.inner.record(&stream.inner)?)
    }

    pub fn synchronize(&self) -> Result<()> {
        Ok(self.inner.synchronize()?)
    }

    pub fn elapsed_ms(&self, end: &Self) -> Result<f32> {
        Ok(self.inner.elapsed_ms(&end.inner)?)
    }
}

impl Stream {
    pub fn wait(&self, event: &Event) -> Result<()> {
        Ok(self.inner.wait(&event.inner)?)
    }
}