From 4f409de0e92fb5f7b44570cce410e4fba1f76ee1 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Wed, 26 Jun 2024 17:01:43 -0500 Subject: [PATCH] Type inference: Use np.bool_ where appropriate, not int32 --- loopy/type_inference.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/loopy/type_inference.py b/loopy/type_inference.py index 8c0aabda5..2919ca798 100644 --- a/loopy/type_inference.py +++ b/loopy/type_inference.py @@ -522,22 +522,20 @@ class TypeInferenceMapper(CombineMapper): return [NumpyType(dtype)] def map_comparison(self, expr): - # "bool" is unusable because OpenCL's bool has indeterminate memory - # format. self(expr.left, return_tuple=False, return_dtype_set=False) self(expr.right, return_tuple=False, return_dtype_set=False) - return [NumpyType(np.dtype(np.int32))] + return [NumpyType(np.dtype(np.bool_))] def map_logical_not(self, expr): self.rec(expr.child) - return [NumpyType(np.dtype(np.int32))] + return [NumpyType(np.dtype(np.bool_))] def map_logical_and(self, expr): for child in expr.children: self.rec(child) - return [NumpyType(np.dtype(np.int32))] + return [NumpyType(np.dtype(np.bool_))] map_logical_or = map_logical_and -- GitLab