dspk-ops 0.0.6

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

//! Filling a buffer with one value on the device.

use cubecl::prelude::*;
use dspk_core::compute::element_position;

use crate::core::DspFloat;

/// `out[i] = value` for `i < len`. One unit per element.
#[cube(launch)]
pub fn fill_kernel<F: DspFloat>(out: &mut [F], value: F, len: u32) {
    let i = element_position() as u32;
    if i < len {
        out[i as usize] = value;
    }
}