torsh-fx 0.1.2

Graph-based model representation and transformation for ToRSh
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

//! Core types and data structures for the FX graph system

use serde::{Deserialize, Serialize};
use std::collections::HashMap;

/// Graph node
#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum Node {
    Input(String),
    Call(String, Vec<String>),
    Output,
    /// Conditional node: if condition, then_branch, else_branch
    Conditional {
        condition: String,
        then_branch: Vec<String>,
        else_branch: Vec<String>,
    },
    /// Loop node: condition, body, loop_vars
    Loop {
        condition: String,
        body: Vec<String>,
        loop_vars: Vec<String>,
    },
    /// Merge node for control flow convergence
    Merge {
        inputs: Vec<String>,
    },
    /// GetAttr node for attribute access
    GetAttr {
        target: String,
        attr: String,
    },
}

/// Graph edge
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Edge {
    pub name: String,
}

/// Detailed statistics about graph structure
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GraphStats {
    pub total_nodes: usize,
    pub total_edges: usize,
    pub input_count: usize,
    pub output_count: usize,
    pub max_fanout: usize,
    pub max_fanin: usize,
    pub average_fanout: f64,
    pub depth: usize,
    pub is_linear: bool,
    pub is_pipeline: bool,
    pub has_cycles: bool,
    pub complexity_score: f64,
    pub node_type_distribution: HashMap<String, usize>,
    pub operation_distribution: HashMap<String, usize>,
    pub fanout_distribution: HashMap<usize, usize>,
    pub fanin_distribution: HashMap<usize, usize>,
}

/// Memory usage estimation for a graph
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MemoryEstimate {
    pub total_bytes: usize,
    pub nodes_bytes: usize,
    pub edges_bytes: usize,
    pub metadata_bytes: usize,
    pub overhead_bytes: usize,
    pub estimated_peak_multiplier: f64,
}