def get_dtype_registry(self):
try:
from pyopencl.compyte.dtypes import TYPE_REGISTRY
except ImportError:
result = _LegacyTypeRegistryStub()
else:
result = TYPE_REGISTRY
from loopy.target.opencl import DTypeRegistryWrapperWithCL1Atomics
if self.atomics_flavor == "cl1":
return DTypeRegistryWrapperWithCL1Atomics(result)
else:
raise NotImplementedError("atomics flavor: %s" % self.atomics_flavor)
def get_dtype_registry(self):
from pyopencl.compyte.dtypes import TYPE_REGISTRY
result = TYPE_REGISTRY
from loopy.target.opencl import (DTypeRegistryWrapperWithCL1Atomics,
DTypeRegistryWrapperWithInt8ForBool)
result = DTypeRegistryWrapperWithInt8ForBool(result)
if self.atomics_flavor == "cl1":
result = DTypeRegistryWrapperWithCL1Atomics(result)
else:
raise NotImplementedError("atomics flavor: %s" %
self.atomics_flavor)
return result