Skip to main content

Module conv1d

Module conv1d 

Source
Expand description

1D Convolution kernel.

Matches conv1d-kernel-v1.yaml.

Standard 1D convolution with configurable stride and padding. Input layout: c_in x length (row-major). Weight layout: c_out x c_in x kernel_size (row-major). Output layout: c_out x out_length (row-major).

Functions§

conv1d_avx2
AVX2 implementation of 1D convolution.
conv1d_ptx
PTX assembly for the 1D convolution kernel.
conv1d_scalar
Scalar reference implementation of 1D convolution.