1#![cfg_attr(not(feature = "kernel-ir"), no_std)]
2
3#[cfg(feature = "cann")]
4pub mod cann;
5
6pub mod attention;
7
8#[cfg(feature = "tensor-normalization")]
9pub mod normalization;
10
11#[cfg(feature = "tensor-gated-delta")]
12pub mod gated_delta;
13
14#[cfg(feature = "tensor-kda")]
15pub mod kda;
16
17#[cfg(any(feature = "tensor-kda", feature = "tensor-gated-delta"))]
18pub mod sequence_graph;
19
20#[cfg(feature = "tensor-moe")]
21pub mod moe;
22
23#[cfg(feature = "kernel-ir")]
24pub mod convolution;
25
26#[cfg(feature = "pooling")]
27pub mod pooling;
28
29#[cfg(any(feature = "pooling", feature = "interpolation"))]
30mod indexing;
31
32#[cfg(feature = "interpolation")]
33pub mod interpolation;
34
35#[cfg(feature = "grid-sample")]
36pub mod grid_sample;
37
38#[cfg(feature = "ctc")]
39pub mod ctc;
40
41
42
43
44
45#[cfg(feature = "tensor-paged-attention")]
46pub mod paged_attention;
47
48#[cfg(feature = "tensor-paged-attention")]
49pub mod dense_attention;
50#[cfg(feature = "tensor-paged-attention")]
51pub mod dense_attention_split;
52#[cfg(feature = "tensor-paged-attention")]
53pub mod dense_attention_rounded;
54
55#[cfg(feature = "tensor-paged-attention")]
56pub mod flash_training;
57
58#[cfg(feature = "tensor-paged-attention")]
59pub mod varlen_attention;
60#[cfg(feature = "tensor-paged-attention")]
61pub mod ring_attention;