From e2f3d6fa941e7e1b9fd222c0a181f4ba3a7df72d Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 6 Mar 2022 15:53:37 -0600
Subject: [PATCH] Type inference for np.generic: Return the specified dtype
---
loopy/type_inference.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/loopy/type_inference.py b/loopy/type_inference.py
index 48324c424..804ab7a3f 100644
--- a/loopy/type_inference.py
+++ b/loopy/type_inference.py
@@ -355,6 +355,8 @@ class TypeInferenceMapper(CombineMapper):
return self.combine([n_dtype_set, d_dtype_set])
def map_constant(self, expr):
+ if isinstance(expr, np.generic):
+ return [NumpyType(np.dtype(type(expr)))]
if is_integer(expr):
for tp in [np.int32, np.int64]:
iinfo = np.iinfo(tp)
--
GitLab