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
43
44

use std::sync::Arc;

use cudarc::driver::{CudaContext, profiler_start, profiler_stop};

use super::Context;
use crate::Result;

#[derive(Debug)]
pub struct ProfilerRange {
    context: Arc<CudaContext>,
    active: bool,
}

impl Context {
    pub fn start_profiler_range(&self) -> Result<ProfilerRange> {
        self.inner.bind_to_thread()?;
        profiler_start()?;
        Ok(ProfilerRange {
            context: Arc::clone(&self.inner),
            active: true,
        })
    }
}

impl ProfilerRange {
    pub fn stop(mut self) -> Result<()> {
        self.context.bind_to_thread()?;
        profiler_stop()?;
        self.active = false;
        Ok(())
    }
}

impl Drop for ProfilerRange {
    fn drop(&mut self) {
        if !self.active {
            return;
        }
        match self.context.bind_to_thread() {
            Ok(()) => self.context.record_err(profiler_stop()),
            Err(error) => self.context.record_err::<()>(Err(error)),
        }
    }
}