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