runmat-runtime 0.4.1

Core runtime for RunMat with builtins, BLAS/LAPACK integration, and execution APIs
Documentation 
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 runmat_thread_local::runmat_thread_local;
use std::cell::RefCell;
use std::sync::{
    atomic::{AtomicBool, Ordering},
    Arc,
};

runmat_thread_local! {
    static INTERRUPT_HANDLE: RefCell<Option<Arc<AtomicBool>>> = const { RefCell::new(None) };
}

pub struct InterruptGuard {
    previous: Option<Arc<AtomicBool>>,
}

impl InterruptGuard {
    pub fn install(handle: Option<Arc<AtomicBool>>) -> Self {
        let previous = INTERRUPT_HANDLE.with(|slot| slot.replace(handle));
        Self { previous }
    }
}

impl Drop for InterruptGuard {
    fn drop(&mut self) {
        INTERRUPT_HANDLE.with(|slot| {
            slot.replace(self.previous.take());
        });
    }
}

pub fn replace_interrupt(handle: Option<Arc<AtomicBool>>) -> InterruptGuard {
    InterruptGuard::install(handle)
}

pub fn is_cancelled() -> bool {
    INTERRUPT_HANDLE.with(|slot| {
        slot.borrow()
            .as_ref()
            .map(|flag| flag.load(Ordering::Relaxed))
            .unwrap_or(false)
    })
}