dspk-ops 0.0.8

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

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

/// One function of each sample (`op`, fixed at compile time): `0` `|x|`, `1` `x²`, `2` `√max(x, 0)`,
/// `3` `log₁₀ max(|x|, floor)`.
#[cube(launch)]
pub fn elementwise_kernel<F: Float + CubeElement + LaunchArg>(input: &[F], output: &mut [F], floor: F, #[comptime] op: u32) {
    let i = element_position();
    if i < input.len() {
        let x = input[i];
        if comptime!(op == 0) {
            output[i] = F::abs(x);
        } else if comptime!(op == 1) {
            output[i] = x * x;
        } else if comptime!(op == 2) {
            output[i] = F::sqrt(F::max(x, F::new(0.0f32)));
        } else {
            output[i] = F::ln(F::max(F::abs(x), floor)) * F::new(core::f32::consts::LOG10_E);
        }
    }
}