Skip to content
Snippets Groups Projects
Commit 38f05a83 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Py3 fix

parent d7223eea
No related branches found
No related tags found
No related merge requests found
......@@ -22,13 +22,12 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
"""
from six.moves import reduce
from pymbolic.mapper import \
IdentityMapper, \
CSECachingMapperMixin
class ConstantFoldingMapperBase(object):
def is_constant(self, expr):
from pymbolic.mapper.dependency import DependencyMapper
......@@ -75,7 +74,6 @@ class ConstantFoldingMapperBase(object):
return self.fold(expr, Sum, operator.add, flattened_sum)
class CommutativeConstantFoldingMapperBase(ConstantFoldingMapperBase):
def map_product(self, expr):
from pymbolic.primitives import Product, flattened_product
......@@ -84,16 +82,15 @@ class CommutativeConstantFoldingMapperBase(ConstantFoldingMapperBase):
return self.fold(expr, Product, operator.mul, flattened_product)
class ConstantFoldingMapper(
CSECachingMapperMixin,
ConstantFoldingMapperBase,
CSECachingMapperMixin,
ConstantFoldingMapperBase,
IdentityMapper):
map_common_subexpression_uncached = \
IdentityMapper.map_common_subexpression
class CommutativeConstantFoldingMapper(
CSECachingMapperMixin,
CommutativeConstantFoldingMapperBase,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment