diff --git a/backends/cbackend.py b/backends/cbackend.py index 825463593098881711cc0180bdf99053613a35d3..8560f970ab40e9f05c957cb6d6aa3fe8c487b512 100644 --- a/backends/cbackend.py +++ b/backends/cbackend.py @@ -1,5 +1,9 @@ import sympy as sp -from sympy.utilities.codegen import CCodePrinter +try: + from sympy.utilities.codegen import CCodePrinter +except ImportError: + from sympy.printing.ccode import C99CodePrinter as CCodePrinter + from pystencils.astnodes import Node from pystencils.types import createType, PointerType diff --git a/transformations.py b/transformations.py index afad758d1b600a95013b344ad75977810d0a38d3..fb4683dda2a803e83c0ca2e0ab532713486949d2 100644 --- a/transformations.py +++ b/transformations.py @@ -477,7 +477,7 @@ def typeAllEquations(eqs, typeForSymbol): typedEquations = [] for eq in eqs: - if isinstance(eq, sp.Eq): + if isinstance(eq, sp.Eq) or isinstance(eq, ast.SympyAssignment): newLhs = processLhs(eq.lhs) newRhs = processRhs(eq.rhs) typedEquations.append(ast.SympyAssignment(newLhs, newRhs))