Struct CadStyleEditor 

pub struct CadStyleEditor<'a> { /* private fields */ }

CAD-style topology editor for LLM computation graphs

Implementations

impl<'a> CadStyleEditor<'a>

pub fn new(graph: &'a mut Graph<OperatorType, WeightTensor>) -> Self

Create a new CAD-style editor

Arguments

• graph - Mutable reference to the graph to edit

pub fn with_defaults(graph: &'a mut Graph<OperatorType, WeightTensor>) -> Self

Create editor with default constraints for transformer architectures

pub fn set_auto_save(&mut self, enabled: bool)

Enable or disable auto-save to history

pub fn history(&self) -> &[HistoryEntry]

Get the edit history

pub fn history_len(&self) -> usize

Get the number of history entries

pub fn detect_defects(&self) -> GraphResult<Vec<TopologyDefect>>

Detect topology defects in the graph

Returns

List of detected defects

pub fn add_constraint( &mut self, constraint: TopologyConstraint, ) -> GraphResult<()>

Add a topology constraint

Arguments

• constraint - Constraint to add

pub fn solve_constraints(&mut self) -> GraphResult<ConstraintReport>

Solve all constraints and auto-fix defects

Returns

Constraint validation report

pub fn extract_module(&mut self, path: &str) -> GraphResult<SubGraph>

Extract a module (subgraph) by path

Arguments

• path - Module path (e.g., “layer.0.attention”)

Returns

Extracted subgraph

pub fn replace_module( &mut self, path: &str, new_module: SubGraph, ) -> GraphResult<()>

Replace a module with a new one

Arguments

• path - Module path to replace
• new_module - New module subgraph

pub fn validate_assembly(&self) -> GraphResult<AssemblyReport>

Validate the assembly of modules

Returns

Assembly validation report

pub fn rollback(&mut self, index: usize) -> GraphResult<bool>

Rollback to a specific history entry

Arguments

• index - Index of the history entry to rollback to

Returns

True if rollback was successful

pub fn undo(&mut self) -> GraphResult<bool>

Undo the last operation

Returns

True if undo was successful

pub fn module_cache(&self) -> &HashMap<String, SubGraph>

Get module cache

pub fn validator(&self) -> &TopologyValidator

Get the topology validator

pub fn validator_mut(&mut self) -> &mut TopologyValidator

Get a mutable reference to the validator

pub fn optimize_with_gradients( &mut self, loss_fn: &dyn Fn(&DifferentiableGraph<Vec<f64>>) -> f64, steps: usize, _learning_rate: f64, ) -> GraphResult<OptimizationReport>

Optimize graph structure using gradient descent on DifferentiableGraph

This method integrates DifferentiableGraph with CadStyleEditor, enabling gradient-based architecture search and topology optimization.

Arguments

• loss_fn - Loss function that takes a DifferentiableGraph reference and returns a scalar loss
• steps - Number of optimization steps
• learning_rate - Learning rate for structure updates

Returns

Optimization report with final loss and structure changes

Note

This is a simplified implementation using finite differences for gradient computation. For production use, consider integrating with an autograd framework like dfdx.