diff options
Diffstat (limited to 'symbolic/generator.py')
| -rw-r--r-- | symbolic/generator.py | 6 |
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)) |