reve-rs 0.1.0

REVE EEG Foundation Model — inference in Rust (RLX)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

//! Device setup hooks for RLX backends.

/// Configure RLX env overrides for known backend quirks before compiling.
pub fn prepare_device(device: rlx::Device) {
    if !matches!(device, rlx::Device::Metal) {
        return;
    }

    // Unfuse elementwise regions on deep graphs (MSL `ElementwiseRegion` bug).
    if rlx::ir::env::is_unset("RLX_METAL_UNFUSE_REGIONS")
        && rlx::ir::env::is_unset("RLX_METAL_NO_FUSION")
    {
        rlx::ir::env::set("RLX_METAL_UNFUSE_REGIONS", "1");
    }
}