diff --git a/loopy/target/pyopencl.py b/loopy/target/pyopencl.py index 2952964442b5fd5b45f54b282deec9e1ac26b3fe..2fd6af935a1a808f0ef01812b743420169ae7704 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