libmir-cuda 0.1.0

CUDA inference backend for libmir
1
2
3
4
5
6
7
8
9
mod gated;
mod pair;
mod reduce;

pub use gated::{
    GatedActivation, SelectedAffineGated, SelectedAffineGatedLaunch, SelectedAffineGatedSpec,
};
pub use pair::{SelectedAffinePair, SelectedAffinePairLaunch, SelectedAffinePairSpec};
pub use reduce::{SelectedAffineReduce, SelectedAffineReduceLaunch, SelectedAffineReduceSpec};