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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

use core::cell::Cell;
use std::thread_local;

thread_local! {
    pub(crate) static COUNT: Cell<usize> = Cell::new(0);
}

/// Sets current cache identifier / index.
/// This function is usually called after an iteration in a loop -> [Count](crate::Count) or [range](crate::range)
/// # Safety
/// Manually setting the count may yield multiple `Buffer` pointing two the same data.
#[inline]
pub unsafe fn set_count(count: usize) {
    COUNT.with(|c| c.set(count));
}

/// Returns current cache identifier / index
#[inline]
pub fn get_count() -> usize {
    COUNT.with(|c| c.get())
}

#[inline]
/// Increases the cache identifier / index by 1.
pub fn bump_count() {
    COUNT.with(|c| {
        let count = c.get();
        c.set(count + 1);
    })
}

#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq, PartialOrd, Ord)]
/// An `Ident` is used to identify a cached pointer.
pub struct Ident {
    /// The index of the `Ident`.
    pub idx: usize,
    /// The amount of elements a corresponding [`Buffer`](crate::Buffer) has.
    pub len: usize,
}

impl Ident {
    /// Returns a new `Ident` with the current cache identifier / index.
    #[inline]
    pub fn new(len: usize) -> Ident {
        Ident {
            idx: get_count(),
            len,
        }
    }

    /// Returns a new `Ident` with the current cache identifier / index and increases the cache identifier / index by 1.
    #[inline]
    pub fn new_bumped(len: usize) -> Ident {
        let id = Ident {
            idx: get_count(),
            len,
        };
        bump_count();
        id
    }
}