candle_ug/
lib.rs

1//! This crate is used to re-export the `ug` crate together with `ug-cuda` & `ug-metal` gated
2//! behind the `cuda` and `metal` features respectively.
3
4pub use ug::*;
5
6#[cfg(feature = "cuda")]
7pub mod cuda {
8    pub use ug_cuda::*;
9}
10
11#[cfg(feature = "metal")]
12pub mod metal {
13    pub use ug_metal::*;
14}