deep_causality_tensor 0.2.2

Tensor data structure for for deep_causality crate.
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

/*
 * SPDX-License-Identifier: MIT
 * Copyright (c) "2025" . The DeepCausality Authors and Contributors. All Rights Reserved.
 */

//! # deep_causality_tensor
//!
//! Hardware-agnostic tensor library with backend abstraction for CPU and GPU computation.

mod errors;
mod extensions;
mod traits;
mod types;
mod utils;

// === Errors ===
pub use crate::errors::causal_tensor_error::CausalTensorError;
pub use crate::errors::ein_sum_validation_error::EinSumValidationError;

// === Traits ===
pub use crate::traits::backend_linear_algebra::LinearAlgebraBackend;
pub use crate::traits::backend_tensor::TensorBackend;
pub use crate::traits::tensor::Tensor;
pub use crate::traits::tensor_data::TensorData;

// === Types ===
pub use crate::types::backend_tensor::BackendTensor;
pub use crate::types::cpu_tensor::InternalCpuTensor;
pub use crate::types::cpu_tensor::{EinSumAST as GenericEinSumAST, EinSumOp as GenericEinSumOp};

// === Backend Types ===
pub use crate::types::backend::Device;
pub use crate::types::backend::cpu::CpuBackend;
#[cfg(all(feature = "mlx", target_os = "macos", target_arch = "aarch64"))]
pub use crate::types::backend::mlx::{MlxBackend, MlxCausalTensor};

// === Extensions ===
pub use crate::extensions::ext_hkt::CausalTensorWitness;
pub use crate::extensions::ext_math::CausalTensorMathExt;

// === Utils (test support) ===
pub use crate::utils::utils_tests;

// === Type Aliases ===

/// Default backend for CausalTensor.
/// Defaults to CPU. Use `mlx` feature for MLX backend (when available).
#[cfg(not(feature = "mlx"))]
pub type DefaultBackend = CpuBackend;

#[cfg(feature = "mlx")]
pub type DefaultBackend = CpuBackend; // Fallback until MlxBackend is ready

/// Default floating-point precision matching the backend.
/// - `mlx` feature: `f32` (GPU native precision)
/// - Default: `f64` (Full precision for physics verification)
#[cfg(all(feature = "mlx", target_os = "macos", target_arch = "aarch64"))]
pub type DefaultFloat = f32;

#[cfg(not(all(feature = "mlx", target_os = "macos", target_arch = "aarch64")))]
pub type DefaultFloat = f64;

/// Public CausalTensor type, generic over backing data but using DefaultBackend.
pub type CausalTensor<T> = BackendTensor<T, DefaultBackend>;

/// Public EinSumOp, generic over T, matches CausalTensor<T>.
pub type EinSumOp<T> = GenericEinSumOp<CausalTensor<T>>;

/// Public EinSumAST, generic over T, matches CausalTensor<T>.
pub type EinSumAST<T> = GenericEinSumAST<CausalTensor<T>>;