diff --git a/loopy/expression.py b/loopy/expression.py index 9d48a8f0073f3430f645087f7ba330b0112c6005..5873e8ea8f43c13d6fbe0f14535790b4a293b5d1 100644 --- a/loopy/expression.py +++ b/loopy/expression.py @@ -176,11 +176,8 @@ class VectorizabilityChecker(Mapper[bool, []]): return False - def map_comparison(self, expr): - # FIXME: These actually can be vectorized: - # https://www.khronos.org/registry/cl/sdk/1.0/docs/man/xhtml/relationalFunctions.html - - raise UnvectorizableError() + def map_comparison(self, expr: p.Comparison) -> bool: + return any(self.rec(child) for child in [expr.left, expr.right]) def map_logical_not(self, expr: object) -> bool: raise UnvectorizableError()