TypeInferenceMapper: allow np.bool in map_type_case

inducer · Jul 19, 2024 · 07cee43 · 07cee43
1 parent da5c786
commit 07cee43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/loopy/type_inference.py b/loopy/type_inference.py
@@ -396,7 +396,7 @@ def map_constant(self, expr):
 
     def map_type_cast(self, expr):
         subtype, = self.rec(expr.child)
-        if not issubclass(subtype.dtype.type, np.number):
+        if not issubclass(subtype.dtype.type, (np.number, np.bool_)):
             raise LoopyError(f"Can't cast a '{subtype}' to '{expr.type}'")
         return [expr.type]