Struct ReflectionPad1dBackward 

pub struct ReflectionPad1dBackward<D: Num> {
    pub name: &'static str,
    pub id: String,
    pub pad_left: i32,
    pub pad_right: i32,
    pub block_ol: i32,
    pub kernel_source: String,
    pub entry_point_source: String,
    pub source: String,
    /* private fields */
}

1-D reflection padding backward pass.

Each output gradient position maps back to one input position via the same reflection rule. Multiple output positions may map to the same input position (the boundary elements reflect), so atomic_add is used. dx must be zero-initialised before launch.

Fields

name: &'static str

The kernel function’s name (e.g. "flash_attention2_forward").

id: String

Unique kernel identifier: fn_name + dtype(s) + const values joined by “__”.

pad_left: i32

Compile-time kernel constant, from the annotated fn’s const generics.

pad_right: i32

Compile-time kernel constant, from the annotated fn’s const generics.

block_ol: i32

Compile-time kernel constant, from the annotated fn’s const generics.

kernel_source: String

The original kernel function source.

entry_point_source: String

The Rust source of the generated C-ABI entry-point wrapper function.

source: String

Combined source (kernel_source + "\n\n" + entry_point_source); used by the Kernel trait.

Implementations

impl<D: Num> ReflectionPad1dBackward<D>

pub fn new(pad_left: i32, pad_right: i32, block_ol: i32) -> Self

Constructs a new kernel instance for these compile-time parameters.

Trait Implementations

impl<D: Num> Kernel for ReflectionPad1dBackward<D>

type Args<'__a> = (*mut D, *mut D, i32, i32, i32, i32)

This kernel’s launch-argument tuple type.

fn id(&self) -> String

A content hash of this kernel’s source, used as a cache key.

fn name(&self) -> &str

This kernel’s name.

fn source(&self) -> &str

This kernel’s full source (DSL + kernel body + entry point wrapper).

fn kernel_source(&self) -> &str

This kernel’s body source, without the entry-point wrapper.

fn entry_point_source(&self) -> &str

Returns the Rust source of the generated C-ABI entry-point wrapper function.

fn entry_point_name(&self) -> String

Returns the PTX symbol name for this kernel: "{name}_entry_point".