dspk-ops 0.0.5

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 dspk_core::compute::{channel_position, sample_position};

/// Spatial projection with at most `width` nonzero weights per output channel (ELLPACK rows):
/// `output[o, t] = Σ_k values[o, k] · input[indices[o, k], t]`. Padding entries carry weight 0.
/// One unit per `(output channel, sample)`.
#[cube(launch)]
pub fn sparse_rows_multiply_kernel<F: Float>(
    input: &[F],
    values: &[F],
    indices: &[u32],
    output: &mut [F],
    num_channels: u32,
    num_samples: u32,
    width: u32,
) {
    let t = sample_position();
    let out_ch = channel_position();

    if t < num_samples && out_ch < num_channels {
        let row = out_ch * width;
        let mut acc = F::new(0.0f32);
        let mut k = 0u32;
        while k < width {
            let in_ch = indices[(row + k) as usize];
            acc += values[(row + k) as usize] * input[(in_ch * num_samples + t) as usize];
            k += 1u32;
        }
        output[(out_ch * num_samples + t) as usize] = acc;
    }
}