From 856b8226d84b3f86c20b974f2994f15719342c98 Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Wed, 26 Jul 2017 15:03:36 +0200
Subject: [PATCH] sympy 1.1 compatibility
---
backends/cbackend.py | 6 +++++-
transformations.py | 2 +-
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/backends/cbackend.py b/backends/cbackend.py
index 825463593..8560f970a 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 afad758d1..fb4683dda 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))
--
GitLab