compiler: Check sympy_type returns a floating point type

devitocodes · Jan 10, 2025 · 457d716 · 457d716
1 parent 71e7eda
commit 457d716
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/devito/symbolics/printer.py b/devito/symbolics/printer.py
@@ -44,7 +44,10 @@ def compiler(self):
         return self._settings['compiler']
 
     def single_prec(self, expr=None):
+        # Extract the dtype of the expression
         dtype = sympy_dtype(expr) if expr is not None else self.dtype
+        # Check that the dtype is a floating point type
+        dtype = dtype if np.issubdtype(dtype, np.floating) else self.dtype
         return dtype in [np.float32, np.float16]
 
     def parenthesize(self, item, level, strict=False):

diff --git a/tests/test_symbolics.py b/tests/test_symbolics.py
@@ -313,7 +313,7 @@ def test_cos_vs_cosf():
 
     # Doesn't make much sense, but it's legal
     c = dSymbol('c', dtype=np.int32)
-    assert ccode(cos(c)) == "cos(c)"
+    assert ccode(cos(c)) == "cosf(c)"
 
 
 def test_intdiv():