From 8f3791a0154e9228cfc32e6d8a525f1ca249511f Mon Sep 17 00:00:00 2001 From: Kaushik Kulkarni Date: Fri, 30 Mar 2018 17:24:57 -0500 Subject: [PATCH] Fixes minor error in identifying the NumpyType --- loopy/target/pyopencl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/loopy/target/pyopencl.py b/loopy/target/pyopencl.py index 295296444..2fd6af935 100644 --- a/loopy/target/pyopencl.py +++ b/loopy/target/pyopencl.py @@ -268,8 +268,8 @@ def pyopencl_with_types(in_knl_callable, arg_id_to_dtype): if name in ["real", "imag", "abs"]: return in_knl_callable.copy(name_in_target="%s_%s" % (tpname, name), - arg_id_to_dtype={0: NumpyType(dtype), -1: NumpyType( - dtype.numpy_dtype.type(0).real)}) + arg_id_to_dtype={0: dtype, -1: NumpyType( + np.dtype(dtype.numpy_dtype.type(0).real))}) return None -- GitLab