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

#[cfg(target_arch = "wasm32")]
use runmat_thread_local::runmat_thread_local;
#[cfg(target_arch = "wasm32")]
use std::cell::RefCell;

#[cfg(target_arch = "wasm32")]
runmat_thread_local! {
    static REGISTRATIONS: RefCell<Vec<Box<dyn FnOnce()>>> = RefCell::new(Vec::new());
}

#[cfg(target_arch = "wasm32")]
pub fn register(f: Box<dyn FnOnce()>) {
    REGISTRATIONS.with(|queue| queue.borrow_mut().push(f));
}

#[cfg(target_arch = "wasm32")]
pub fn flush() {
    REGISTRATIONS.with(|queue| {
        for f in queue.borrow_mut().drain(..) {
            f();
        }
    });
}

#[cfg(not(target_arch = "wasm32"))]
pub fn register(_f: Box<dyn FnOnce()>) {}

#[cfg(not(target_arch = "wasm32"))]
pub fn flush() {}