Struct FlattenBackward 

pub struct FlattenBackward<D: Num> {
    pub name: &'static str,
    pub id: String,
    pub block_b: i32,
    pub block_n: i32,
    pub kernel_source: String,
    pub entry_point_source: String,
    pub source: String,
    /* private fields */
}

Copy a contiguous row-major [B, N] gradient back to an output buffer with arbitrary strides.

This is the backward pass of the flatten operation. The gradient dy arrives as a contiguous row-major tensor (the upstream gradient matches the contiguous forward output); this kernel writes it back to dx using the original input strides so that the gradient is in the same memory layout as the forward input. When the forward input was already row-major contiguous this is again a simple memcpy; when the forward input used a different layout the kernel performs the inverse reordering.

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 “__”.

block_b: i32

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

block_n: 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> FlattenBackward<D>

pub fn new(block_b: i32, block_n: i32) -> Self

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

Trait Implementations

impl<D: Num> Kernel for FlattenBackward<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".