Gather interop moments in a more generic manner

i.e. return unshifted moments in a implicitly ordered float4 array.
Cell positions are reconstructed by a vertex shaded analogously to
how it is done in compustream.

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))