hanzo-rocm-kernels 0.10.2

ROCm/HIP kernels for Hanzo
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

pub mod compile;
pub mod error;
pub mod kernel;
pub mod ops;
pub mod utils;
pub mod wrappers;

pub use compile::KernelCache;
pub use error::KernelError;
pub use kernel::{
    AffineKernel, BinaryKernel, BinaryOp, CastKernel, DType, FillKernel, IndexingKernel,
    KernelSource, ReduceKernel, UnaryKernel, UnaryOp,
};
pub use ops::OpLauncher;
pub use utils::BufferOffset;