Struct AttentionOptions 

pub struct AttentionOptions {
    pub causal: bool,
    pub causal_offset: u32,
    pub window: u32,
    pub dropout_prob: f32,
    pub seed: u64,
}

Options for ComputeEngine::flash_attention_v2_with and ComputeEngine::flash_attention_v2_backward_with.

Query i sits at absolute position i + causal_offset; key j is visible when j <= i + causal_offset (if causal) and |i + causal_offset - j| < window (if window > 0).

Fields

causal: bool

Hide keys after each query’s absolute position.

causal_offset: u32

Absolute position of the first query, e.g. the number of tokens already in a KV cache when decoding.

window: u32

Sliding-window size in tokens; 0 means unlimited.

dropout_prob: f32

Attention dropout probability; 0.0 disables dropout.

seed: u64

Seed for the dropout pattern (only the low 32 bits are used).

Trait Implementations

impl Clone for AttentionOptions

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for AttentionOptions

impl Debug for AttentionOptions

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

Formats the value using the given formatter.

impl Default for AttentionOptions

fn default() -> Self

Returns the “default value” for a type.