From 27cf4f19418e1135be306bee09ba9a7b51b60051 Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Mon, 14 May 2018 17:32:39 +0200
Subject: [PATCH] Fixes in vectorization to also support float kernels
---
backends/cbackend.py | 1 +
transformations.py | 2 ++
2 files changed, 3 insertions(+)
diff --git a/backends/cbackend.py b/backends/cbackend.py
index 76eb3ee74..00da70f27 100644
--- a/backends/cbackend.py
+++ b/backends/cbackend.py
@@ -272,6 +272,7 @@ class CustomSympyPrinter(CCodePrinter):
else:
return res
+
# noinspection PyPep8Naming
class VectorizedCustomSympyPrinter(CustomSympyPrinter):
SummandInfo = namedtuple("SummandInfo", ['sign', 'term'])
diff --git a/transformations.py b/transformations.py
index ab7d98ab5..d108b7011 100644
--- a/transformations.py
+++ b/transformations.py
@@ -721,6 +721,8 @@ class KernelConstraintsCheck:
elif isinstance(rhs, sp.Mul):
new_args = [self.process_expression(arg) if arg not in (-1, 1) else arg for arg in rhs.args]
return rhs.func(*new_args) if new_args else rhs
+ elif isinstance(rhs, sp.Indexed):
+ return rhs
else:
if isinstance(rhs, sp.Pow):
# don't process exponents -> they should remain integers
--
GitLab