iris-cv 0.0.0

A fast computer vision library framework in Rust.
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

use burn::tensor::backend::Backend;

/// Utility helpers for configuring and running Burn operations.
pub struct BurnUtils;

impl BurnUtils {
    #[must_use]
    pub fn backend_name<B: Backend>() -> String {
        let device = Default::default();
        B::name(&device)
    }

    /// Selects the best device available for this execution.
    #[must_use]
    pub fn best_device<B: Backend>() -> B::Device {
        // Defaults to WGPU or CPU device based on the backend used.
        Default::default()
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::test_helpers::TestBackend;

    #[test]
    fn test_burn_utils() {
        let name = BurnUtils::backend_name::<TestBackend>();
        assert!(!name.is_empty());
        let _device = BurnUtils::best_device::<TestBackend>();
    }
}