cubek-attention 0.2.0

CubeK: Attention Kernels
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
#![allow(clippy::explicit_counter_loop, clippy::type_complexity)]

/// Components for matrix multiplication
pub mod components;
pub mod launch;
/// Contains attention kernels
pub mod routines;

pub mod definition;

#[cfg(feature = "cpu-reference")]
pub mod cpu_reference;