ruda-tensor-device 0.21.20

Ruda device tensor backend and domain-library dispatch.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

#![warn(missing_docs)]
#![cfg_attr(docsrs, feature(doc_cfg))]

//! Ruda device tensor backend and domain-library dispatch.

#[macro_use]
extern crate derive_new;
extern crate alloc;

/// Backend implementations of tensor operations.
pub mod dispatch;
/// Scalar element contracts for device tensor operations.
pub mod element;
mod backend;
mod runtime;

pub use backend::DeviceBackend;
pub use runtime::DeviceRuntime;
pub use element::{BoolElement, TensorElement, FloatElement, IntElement};
pub use ruda_kernel::tensor::RudaTensor;

#[cfg(feature = "fusion")]
/// Device backend integration with tensor fusion.
pub mod fusion;

/// NVIDIA CUDA device tensor adapter.
#[cfg(feature = "cuda")]
pub mod cuda;