Struct FlashConfig 

pub struct FlashConfig {
    pub block_size_q: usize,
    pub block_size_kv: usize,
    pub causal: bool,
    pub dropout_p: f32,
}

Configuration for FlashAttention tiled computation.

Fields

block_size_q: usize

Block size along the query dimension (Br).

block_size_kv: usize

Block size along the key/value dimension (Bc).

causal: bool

Whether to apply causal masking (upper-triangular mask).

dropout_p: f32

Dropout probability (0.0 = no dropout). Applied conceptually but not stochastically in this CPU implementation.

Implementations

impl FlashConfig

pub fn new(block_size_q: usize, block_size_kv: usize) -> AttentionResult<Self>

Creates a config with custom block sizes.

pub fn with_causal(self) -> Self

Returns a causal variant of this config.

pub fn with_dropout(self, p: f32) -> AttentionResult<Self>

Sets the dropout probability.

Trait Implementations

impl Clone for FlashConfig

fn clone(&self) -> FlashConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for FlashConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for FlashConfig

fn default() -> Self

Returns the “default value” for a type.