diff --git a/loopy/expression.py b/loopy/expression.py index 5873e8ea8f43c13d6fbe0f14535790b4a293b5d1..5a11b835474cb0933ffaf623099ce0c4b5f0b937 100644 --- a/loopy/expression.py +++ b/loopy/expression.py @@ -189,6 +189,9 @@ class VectorizabilityChecker(Mapper[bool, []]): # FIXME: Do this more carefully raise UnvectorizableError() + def map_if(self, expr: p.If) -> bool: + return any(self.rec(child) for child in [expr.condition, expr.then, expr.else_]) + # }}} # vim: fdm=marker