compiler: Patch unexpansion with double precision

devitocodes · Sep 21, 2023 · a96a803 · a96a803
1 parent b23d8ed
commit a96a803
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/devito/passes/clusters/derivatives.py b/devito/passes/clusters/derivatives.py
@@ -74,7 +74,7 @@ def _core(expr, c, weights, mapper, sregistry):
     try:
         w = weights[k]
     except KeyError:
-        w = weights[k] = w0._rebuild(name=name)
+        w = weights[k] = w0._rebuild(name=name, dtype=expr.dtype)
     expr = uxreplace(expr, {w0.indexed: w.indexed})
 
     dims = retrieve_dimensions(expr, deep=True)

diff --git a/tests/test_unexpansion.py b/tests/test_unexpansion.py
@@ -280,6 +280,20 @@ def test_v1(self):
         op.cfunction
 
 
+class TestMisc(object):
+
+    def test_double_precision(self):
+        grid = Grid(shape=(10, 10, 10), dtype=np.float64)
+
+        u = TimeFunction(name='u', grid=grid, space_order=4)
+
+        eqns = Eq(u.forward, u.laplace + 1.)
+
+        op = Operator(eqns, opt=('advanced', {'expand': False}))
+
+        op.cfunction
+
+
 def tti_sa_eqns(grid):
     t = grid.stepping_dim
     x, y, z = grid.dimensions