TypeInferenceMapper: allow np.bool in map_type_case

illinois-ceesd · Jul 18, 2024 · f494955 · f494955
1 parent 7d5a1fb
commit f494955
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]