Skip to content
Snippets Groups Projects
Commit 58949541 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Track pyopencl vec -> cltypes switch

parent ebd1bf99
No related branches found
No related tags found
1 merge request!115Fix domain tree branch detection
Pipeline #
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment