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

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

use super::kernels::scale_samples_kernel;
use crate::core::DspFloat;

/// `output = input ยท scale_factor + offset` over `total_elements` values.
pub fn execute_scaling<F: DspFloat>(
    client: &Client,
    input: &cubecl::server::Handle,
    output: &cubecl::server::Handle,
    total_elements: usize,
    scale_factor: F,
    offset: F,
) {
    let geom = LaunchGeometry::elementwise(client, total_elements);
    unsafe {
        scale_samples_kernel::launch::<F>(
            client,
            geom.cube_count,
            geom.cube_dim,
            BufferArg::from_raw_parts(input.clone(), total_elements),
            BufferArg::from_raw_parts(output.clone(), total_elements),
            scale_factor,
            offset,
        );
    }
}