custos 0.7.0

A minimal OpenCL, WGPU, CUDA and host CPU array manipulation engine.
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

#[cfg(not(feature = "realloc"))]
use std::ptr::null_mut;

#[cfg(feature = "cpu")]
#[cfg(not(feature = "realloc"))]
use custos::{range, Buffer, CPU};

#[cfg(feature = "cpu")]
#[cfg(not(feature = "realloc"))]
fn cached_add<'a>(device: &'a CPU, a: &[f32], b: &[f32]) -> Buffer<'a, f32, CPU> {
    let mut out = custos::Device::retrieve::<f32, ()>(device, 10, ());
    for i in 0..out.len() {
        out[i] = a[i] + b[i];
    }
    out
}

#[cfg(feature = "cpu")]
#[cfg(not(feature = "realloc"))]
#[test]
fn test_caching_cpu() {
    let device = CPU::new();

    let a = Buffer::<f32, _>::new(&device, 100);
    let b = Buffer::<f32, _>::new(&device, 100);

    let mut old_ptr = null_mut();

    for _ in range(100) {
        let mut out = cached_add(&device, &a, &b);
        if out.host_ptr() != old_ptr && !old_ptr.is_null() {
            panic!("Should be the same pointer!");
        }
        old_ptr = out.host_ptr_mut();
        let len = device.addons.cache.borrow().nodes.len();
        //let len = CPU_CACHE.with(|cache| cache.borrow().nodes.len());
        assert_eq!(len, 3);
    }
}