enki-gpu 0.2.3

Pure Rust heterogeneous GPU compute platform with JIT compilation
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

pub mod core;
pub mod dispatch;
pub mod format;
pub mod lifecycle;
pub mod ops;
pub mod state;
pub mod transfer;

pub use self::core::GpuVec;
pub use self::state::{BufferPhase, PhaseViolationError};

impl<T: Copy + Send + Sync + 'static> From<GpuVec<T>> for Vec<T> {
    #[inline(always)]
    fn from(gpu_vec: GpuVec<T>) -> Self {
        gpu_vec.to_vec()
    }
}

impl<T: Copy + Send + Sync + 'static> From<&[T]> for GpuVec<T> {
    #[inline(always)]
    fn from(slice: &[T]) -> Self {
        GpuVec::from_slice(slice)
    }
}

impl<T: Copy + Send + Sync + 'static> From<Vec<T>> for GpuVec<T> {
    #[inline(always)]
    fn from(vec: Vec<T>) -> Self {
        GpuVec::from_slice(&vec)
    }
}

/// Creates a [`GpuVec`] containing the provided arguments.
///
/// Matches standard Rust `vec!` syntax:
/// ```rust
/// use enki::gpu_vec;
///
/// // Explicit elements
/// let v1 = gpu_vec![1u32, 2, 3, 4];
///
/// // Repeating elements
/// let v2 = gpu_vec![0.0f32; 1024];
/// ```
#[macro_export]
macro_rules! gpu_vec {
    () => {
        $crate::enki_api::resources::GpuVec::new()
    };
    ($elem:expr; $n:expr) => {
        $crate::enki_api::resources::GpuVec::from_elem($elem, $n)
    };
    ($($x:expr),* $(,)?) => {
        $crate::enki_api::resources::GpuVec::from_slice(&[$($x),*])
    };
}