Struct KernelConfig 

pub struct KernelConfig {
    pub hidden_dim: usize,
    pub num_heads: usize,
    pub head_dim: usize,
    pub seq_len: usize,
    pub use_flash: bool,
    pub memory_format: MemoryFormat,
}

Configuration for sparse attention kernel

Fields

hidden_dim: usize

Hidden dimension

num_heads: usize

Number of heads

head_dim: usize

Head dimension

seq_len: usize

Sequence length

use_flash: bool

Use flash attention

memory_format: MemoryFormat

Memory format (contiguous, channels_last, etc.)

Trait Implementations

impl Clone for KernelConfig

fn clone(&self) -> KernelConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for KernelConfig

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

Formats the value using the given formatter.

impl Default for KernelConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for KernelConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for KernelConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.