diff --git a/loopy/target/pyopencl.py b/loopy/target/pyopencl.py index 467bc8ee801a090641c7e7f8d7f2e7c12a921232..00076ced1d24c880d2340ac8053253b2c29ab29a 100644 --- a/loopy/target/pyopencl.py +++ b/loopy/target/pyopencl.py @@ -363,14 +363,14 @@ class PyOpenCLTarget(OpenCLTarget): raise NotImplementedError("atomics flavor: %s" % self.atomics_flavor) def is_vector_dtype(self, dtype): - from pyopencl.array import vec + import pyopencl.cltypes as cltypes return (isinstance(dtype, NumpyType) - and dtype.numpy_dtype in list(vec.types.values())) + and dtype.numpy_dtype in list(cltypes.vec_types.values())) def vector_dtype(self, base, count): - from pyopencl.array import vec + import pyopencl.cltypes as cltypes return NumpyType( - vec.types[base.numpy_dtype, count], + cltypes.vec_types[base.numpy_dtype, count], target=self) def alignment_requirement(self, type_decl):