burn_cubecl/kernel/
mod.rs

1mod binary;
2mod binary_float;
3mod binary_int;
4mod cast;
5mod clamp;
6mod comparison;
7mod contiguous;
8mod cross;
9mod index;
10mod mask;
11mod unary_float;
12mod unary_int;
13mod unary_numeric;
14
15pub(crate) use binary::*;
16pub(crate) use binary_float::*;
17pub(crate) use binary_int::*;
18pub use cast::*;
19pub use contiguous::*;
20pub(crate) use cross::*;
21pub use mask::*;
22pub(crate) use unary_float::*;
23pub(crate) use unary_int::*;
24pub(crate) use unary_numeric::*;
25
26pub use crate::cubecl::prelude::KernelMetadata;
27
28/// Attention kernels
29pub mod attention;
30/// Convolution kernels
31pub mod conv;
32/// Grid sampling kernels
33pub mod grid_sample;
34/// Interpolation kernels
35pub mod interpolate;
36/// Matmul kernels
37pub mod matmul;
38/// Pooling kernels
39pub mod pool;
40/// Pseudo-random number generator kernels
41pub mod prng;
42/// Quantization operations
43pub mod quantization;
44/// Reduction algorithms
45pub mod reduce;
46
47pub(crate) use clamp::*;
48pub(crate) use comparison::*;
49pub use index::*;
50
51pub(crate) mod utils;
burn_cubecl/kernel/mod.rs

burn_cubecl/kernel/
mod.rs
