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
62
63

//! The WGPU module provides the WGPU backend for custos.
mod launch_shader;
mod shader_cache;
mod wgpu_buffer;
mod wgpu_device;

use core::fmt::Debug;

pub use launch_shader::*;
pub use wgpu_device::*;

use crate::{Buffer, Shape};

/// Sets all the elements of a `WGPU` `Buffer` to zero / default.
///
/// # Example
/// ```
/// use custos::{WGPU, Buffer, wgpu::wgpu_clear};
///
/// fn main() -> custos::Result<()> {
///     let device = WGPU::new(wgpu::Backends::all())?;
///     let mut buf = Buffer::from((&device, [4f32, -1.2, 2., 1., 3.]));
///     wgpu_clear(&device, &mut buf);
///     
///     assert_eq!(buf.read(), [0., 0., 0., 0., 0.,]);
///     Ok(())
/// }
/// ```
pub fn wgpu_clear<T: Default + Debug, S: Shape>(device: &WGPU, buf: &mut Buffer<T, WGPU, S>) {
    let src = format!(
        "@group(0)
        @binding(0)
        var<storage, read_write> buf: array<{datatype}>;
        
        @compute
        @workgroup_size(1)
        fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {{
            buf[global_id.x] = {zero:?};
        }}
        ",
        datatype = std::any::type_name::<T>(),
        zero = T::default()
    );

    launch_shader(device, &src, [buf.len() as u32, 1, 1], &[buf])
}

#[cfg(test)]
mod tests {
    use crate::{Buffer, WGPU};

    #[test]
    fn test_wgpu_clear() -> crate::Result<()> {
        let device = WGPU::new(wgpu::Backends::all())?;

        let mut buf = Buffer::from((&device, [1, 4, 2, 1, 9]));
        buf.clear();

        assert_eq!(buf.read(), [0, 0, 0, 0, 0]);

        Ok(())
    }
}