diff --git a/pycuda/elementwise.py b/pycuda/elementwise.py index 5486d097f57d03d7ec1415a733c7bb216e2fe83d..9d66fe4c237f77324b181c37fe370857d4d688f7 100644 --- a/pycuda/elementwise.py +++ b/pycuda/elementwise.py @@ -463,7 +463,7 @@ def get_linear_combination_kernel(summand_descriptors, dtype_z): return func, tex_src -def _get_real_dtype(dtype: np.dtype[Any]) -> np.dtype[Any]: +def _get_real_dtype(dtype: "np.dtype[Any]") -> "np.dtype[Any]": assert dtype.kind == "c" return np.empty(0, dtype).real.dtype