Start refactor of comp_graph

rileysu · Dec 17, 2023 · b636fa1 · b636fa1
1 parent 6a65e7d
commit b636fa1
Show file tree

Hide file tree

Showing 5 changed files with 123 additions and 189 deletions.
diff --git a/ideas.md b/ideas.md
@@ -29,4 +29,5 @@
     - Use phantom to make tensors references to graph so it can't outlive graph
 - Probably remove the distinction between context and comp_graph
     - Dump graph on calculation
-    - Allow for recalc maybe
+    - Allow for recalc maybe (as in mutating tensors within the graph)
+- Model how the graph interface should look externally as ergonomics is an issue rn
diff --git a/src/context/edge.rs → src/comp_graph/edge.rs b/src/context/edge.rs → src/comp_graph/edge.rs
@@ -1,6 +1,6 @@
 use crate::engine::{tensor::{allowed_unit::AllowedUnit, EngineTensor}, EngineError};
 
-use super::comp_graph::{NodeKey, ComputationGraphError};
+use super::{NodeKey, ComputationGraphError};
 
 #[derive(Clone, Copy, Debug, PartialEq)]
 pub enum Edge<T: AllowedUnit> {