Fix real return type (#728)

FEniCS · Nov 25, 2024 · 7164d8c · 7164d8c
1 parent 712a57d
commit 7164d8c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/ffcx/codegeneration/expression_generator.py b/ffcx/codegeneration/expression_generator.py
@@ -365,7 +365,13 @@ def generate_partition(self, symbol, F, mode):
                 vexpr = L.ufl_to_lnodes(v, *vops)
 
                 is_cond = isinstance(v, ufl.classes.Condition)
-                dtype = L.DataType.BOOL if is_cond else L.DataType.SCALAR
+                is_real = isinstance(v, ufl.classes.Real)
+                if is_cond:
+                    dtype = L.DataType.BOOL
+                elif is_real:
+                    dtype = L.DataType.REAL
+                else:
+                    dtype = L.DataType.SCALAR
 
                 j = len(intermediates)
                 vaccess = L.Symbol(f"{symbol.name}_{j}", dtype=dtype)