From 58949541fe9b24824aa96a35a3eb102dc1b59db6 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 2 Jun 2017 13:11:00 -0400
Subject: [PATCH] Track pyopencl vec -> cltypes switch
---
loopy/target/pyopencl.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/loopy/target/pyopencl.py b/loopy/target/pyopencl.py
index 467bc8ee8..00076ced1 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):
--
GitLab