ghostflow-core 1.1.0

Core tensor operations for GhostFlow ML framework - optimized for maximum performance
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
71
72
73

//! GhostFlow Core - High-performance tensor operations
//! 
//! This crate provides the foundational tensor type and operations
//! for the GhostFlow ML framework.
//!
//! ## Phase 4 Optimizations (Beat JAX!)
//! - Operation fusion engine
//! - JIT compilation
//! - Memory layout optimization
//! - Custom optimized kernels

pub mod dtype;
pub mod shape;
pub mod storage;
pub mod tensor;
pub mod ops;
pub mod device;
pub mod error;
pub mod serialize;
pub mod sparse;
pub mod hardware;
pub mod rocm;
pub mod metal;
pub mod neon;
pub mod tpu;

// Phase 4: Advanced optimizations
pub mod fusion;
// pub mod jit; // Temporarily disabled - needs refactoring
pub mod layout;

// Performance optimizations
pub mod simd_ops;
pub mod memory;
pub mod profiler;

pub use dtype::DType;
pub use shape::{Shape, Strides};
pub use storage::Storage;
pub use tensor::Tensor;
pub use device::{Device, Cpu};
pub use error::{GhostError, Result};
pub use serialize::{StateDict, save_state_dict, load_state_dict, Serializable};
pub use sparse::{SparseTensorCOO, SparseTensorCSR, SparseTensorCSC};
pub use hardware::{HardwareBackend, HardwareDevice, HardwareOps, ElementwiseOp, list_devices};

// Phase 4 exports
pub use fusion::{FusionEngine, ComputeGraph, FusionPattern};
// pub use jit::{JitCompiler, CompiledKernel}; // Temporarily disabled - needs refactoring
pub use layout::{LayoutOptimizer, MemoryLayout, DeviceInfo};

// Performance exports
pub use simd_ops::{simd_add_f32, simd_mul_f32, simd_dot_f32, simd_relu_f32};
pub use memory::{MemoryPool, MemoryStats, MemoryLayoutOptimizer, TrackedAllocator};
pub use profiler::{Profiler, ProfileScope, Benchmark, BenchmarkResult, global_profiler};

/// Prelude for convenient imports
#[allow(unused_imports)]
pub mod prelude {
    pub use crate::{Tensor, DType, Shape, Device, Cpu};
    pub use crate::tensor_ops::*;
    pub use crate::serialize::{StateDict, save_state_dict, load_state_dict, Serializable};
    pub use crate::{FusionEngine, LayoutOptimizer};
}

/// Tensor operations trait extensions
#[allow(unused_imports)]
pub mod tensor_ops {
    pub use crate::ops::arithmetic::*;
    pub use crate::ops::reduction::*;
    pub use crate::ops::activation::*;
    pub use crate::ops::matmul::*;
}