From e53c04635f793dd6a8e2d3f4656322f82499578b Mon Sep 17 00:00:00 2001 From: Alexandru Fikl <alexfikl@gmail.com> Date: Wed, 9 Nov 2022 10:27:52 +0200 Subject: [PATCH] tools: fix double precision check --- pyopencl/tools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyopencl/tools.py b/pyopencl/tools.py index 3be064bb..9f7b558c 100644 --- a/pyopencl/tools.py +++ b/pyopencl/tools.py @@ -955,7 +955,7 @@ class _CDeclList: def add_dtype(self, dtype): dtype = np.dtype(dtype) - if dtype in [np.float64 or np.complex128]: + if dtype in (np.float64, np.complex128): self.saw_double = True if dtype.kind == "c": @@ -988,7 +988,7 @@ class _CDeclList: def visit_arguments(self, arguments): for arg in arguments: dtype = arg.dtype - if dtype in [np.float64 or np.complex128]: + if dtype in (np.float64, np.complex128): self.saw_double = True if dtype.kind == "c": -- GitLab