minor refactoring

JuliaSymbolics · Oct 27, 2021 · 918e1cd · 918e1cd · 0x0f0f0f · Oct 27, 2021
1 parent edfa891
commit 918e1cd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/src/EGraphs/ematch.jl b/src/EGraphs/ematch.jl
@@ -41,12 +41,11 @@ function instantiate(g::EGraph, pat::PatTerm, sub::Sub, rule::AbstractRule)
     ar = arity(pat)
 
     T = gettermtype(g, op, ar)
-    println(T)
     children = map(x -> instantiate(g, x, sub, rule), arguments(pat))
     similarterm(T, op, children; exprhead=eh)
 end
 
-
+## ====================== EMatching Machine =======================
 
 mutable struct Machine
     g::EGraph 

diff --git a/src/ematch_compiler.jl b/src/ematch_compiler.jl
@@ -156,7 +156,7 @@ function compile_pat!(reg, p::PatTerm, prog)
     c = memsize(prog)
     nargs = arity(p)
     # registers unit range
-    a = c:(c + nargs - 1)
+    regrange = c:(c + nargs - 1)
 
     exhead = exprhead(p)
     op = operation(p)
@@ -173,9 +173,8 @@ function compile_pat!(reg, p::PatTerm, prog)
 
     increment(prog, nargs)
 
-
-    push!(prog.instructions, Bind(reg, ENodePat(exhead, op, a, checkop)))
-    for (reg, p2) in zip(a, arguments(p))
+    push!(prog.instructions, Bind(reg, ENodePat(exhead, op, regrange, checkop)))
+    for (reg, p2) in zip(regrange, arguments(p))
         compile_pat!(reg, p2, prog) 
     end
 end