morok-device 0.1.0-alpha.2

Device abstraction layer for the Morok ML compiler
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

use crate::DeviceSpec;

#[test]
fn test_registry_cpu() {
    let allocator = crate::cpu().unwrap();
    assert_eq!(allocator.name(), "CPU");
}

#[test]
fn test_max_buffers_cpu() {
    let spec = DeviceSpec::Cpu;
    assert_eq!(spec.max_buffers(), None, "CPU should have no buffer limit");
}

#[cfg(feature = "cuda")]
#[test]
fn test_max_buffers_cuda() {
    let spec = DeviceSpec::Cuda { device_id: 0 };
    assert_eq!(spec.max_buffers(), None, "CUDA should have no buffer limit");
}

#[cfg(feature = "metal")]
#[test]
fn test_max_buffers_metal() {
    let spec = DeviceSpec::Metal { device_id: 0 };
    assert_eq!(spec.max_buffers(), Some(31), "Metal should have 31 buffer limit");
}

#[cfg(feature = "webgpu")]
#[test]
fn test_max_buffers_webgpu() {
    let spec = DeviceSpec::WebGpu;
    assert_eq!(spec.max_buffers(), Some(8), "WebGPU should have 8 buffer limit");
}