ruda_tensor_device/
lib.rs

1#![warn(missing_docs)]
2#![cfg_attr(docsrs, feature(doc_cfg))]
3
4//! Ruda device tensor backend and domain-library dispatch.
5
6#[macro_use]
7extern crate derive_new;
8extern crate alloc;
9
10/// Backend implementations of tensor operations.
11pub mod dispatch;
12/// Scalar element contracts for device tensor operations.
13pub mod element;
14mod backend;
15mod runtime;
16
17pub use backend::DeviceBackend;
18pub use runtime::DeviceRuntime;
19pub use element::{BoolElement, TensorElement, FloatElement, IntElement};
20pub use ruda_kernel::tensor::RudaTensor;
21
22#[cfg(feature = "fusion")]
23/// Device backend integration with tensor fusion.
24pub mod fusion;
25
26/// NVIDIA CUDA device tensor adapter.
27#[cfg(feature = "cuda")]
28pub mod cuda;
ruda_tensor_device/lib.rs

ruda_tensor_device/
lib.rs
