aboutsummaryrefslogtreecommitdiff
path: root/symbolic
diff options
context:
space:
mode:
Diffstat (limited to 'symbolic')
-rw-r--r--symbolic/generator.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/symbolic/generator.py b/symbolic/generator.py
index 853bbf9..f94031f 100644
--- a/symbolic/generator.py
+++ b/symbolic/generator.py
@@ -47,7 +47,7 @@ class LBM:
exprs = [ self.f_curr[i] + 1/tau * (f_eq_i - self.f_curr[i]) for i, f_eq_i in enumerate(f_eq) ]
if optimize:
- helper, f = cse(exprs, optimizations=optimizations.custom)
- return (helper, assign(self.f_next, f))
+ subexprs, f = cse(exprs, optimizations=optimizations.custom)
+ return (subexprs, assign(self.f_next, f))
else:
- return ([], assign(self.f_next, f))
+ return ([], assign(self.f_next, exprs))