dspk-ops 0.0.4

Signal-processing operations for dspk, written once in CubeCL: filters, resampling, linear algebra, statistics, peak finding, spatial operators
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

use cubecl::prelude::*;
use cubecl::server::Handle;

use super::buffer;

/// A device buffer reused across calls, reallocated only when a call needs more than it holds.
#[derive(Debug, Clone, Default)]
pub struct Scratch {
    handle: Option<Handle>,
    capacity_bytes: usize,
}

impl Scratch {
    pub fn new() -> Self {
        Self::default()
    }

    /// A buffer of at least `len` elements of `E` (contents unspecified).
    pub fn get<E: CubeElement>(&mut self, client: &Client, len: usize) -> Handle {
        let need = buffer::bytes::<E>(len);
        match &self.handle {
            Some(handle) if self.capacity_bytes >= need => handle.clone(),
            _ => {
                let handle = client.empty(need);
                self.handle = Some(handle.clone());
                self.capacity_bytes = need;
                handle
            }
        }
    }
}