struct

`SushiBLAS::Kernels::ChannelBiasEpilogue`

Adds a per-channel bias to each GEMM output element, in either layout.

Declared in: include/SushiBLAS/kernels/blas/level3/channel_bias_epilogue.hpp

Warning

The bias pointer must be device-accessible, outlive the submission and be a declared read.

Public attributes

const void * bias = nullptr

The bias vector's first element, as the GEMM's own element type.

int64_t row_stride = 0

Bias elements to advance per output row; 0 when the channel is the column.

int64_t col_stride = 1

Bias elements to advance per output column; 0 when the channel is the row.

template <typename T>
T operator()(T value, int64_t row, int64_t col) const noexcept

Returns value plus the bias for its position.

Parameters