pub struct Conv1dBackwardDw<D: Num> {
pub name: &'static str,
pub id: String,
pub kl: i32,
pub stride: i32,
pub pad: i32,
pub block_ol: i32,
pub kernel_source: String,
pub entry_point_source: String,
pub source: String,
/* private fields */
}Expand description
1-D convolution backward pass — gradient with respect to weights (dw).
Grid: pid = (b * C_OUT + c_out) * num_ol_tiles + ol_tile
dw must be zero-initialised before launch.
Fields§
§name: &'static strThe kernel function’s name (e.g. "flash_attention2_forward").
id: StringUnique kernel identifier: fn_name + dtype(s) + const values joined by “__”.
kl: i32Compile-time kernel constant, from the annotated fn’s const generics.
stride: i32Compile-time kernel constant, from the annotated fn’s const generics.
pad: i32Compile-time kernel constant, from the annotated fn’s const generics.
block_ol: i32Compile-time kernel constant, from the annotated fn’s const generics.
kernel_source: StringThe original kernel function source.
entry_point_source: StringThe Rust source of the generated C-ABI entry-point wrapper function.
source: StringCombined source (kernel_source + "\n\n" + entry_point_source); used by the Kernel trait.
Implementations§
Trait Implementations§
Source§impl<D: Num> Kernel for Conv1dBackwardDw<D>
impl<D: Num> Kernel for Conv1dBackwardDw<D>
Source§type Args<'__a> = (*mut D, *mut D, *mut D, i32, i32, i32, i32, i32)
type Args<'__a> = (*mut D, *mut D, *mut D, i32, i32, i32, i32, i32)
Source§fn kernel_source(&self) -> &str
fn kernel_source(&self) -> &str
Source§fn entry_point_source(&self) -> &str
fn entry_point_source(&self) -> &str
Source§fn entry_point_name(&self) -> String
fn entry_point_name(&self) -> String
"{name}_entry_point".