Python DTypeRegistry.DTypeRegistry Examples

Example #1

File: __init__.py Project: gaohao95/loopy

 def get_dtype_registry(self):
     from loopy.target.c.compyte.dtypes import (
             DTypeRegistry, fill_registry_with_c_types)
     result = DTypeRegistry()
     fill_registry_with_c_types(result, respect_windows=False,
             include_bool=True)
     return DTypeRegistryWrapper(result)

Example #2

File: __init__.py Project: yueyedeai/loopy

 def get_dtype_registry(self):
     from loopy.target.c.compyte.dtypes import (
         DTypeRegistry, fill_registry_with_c99_stdint_types,
         fill_registry_with_c99_complex_types)
     result = DTypeRegistry()
     fill_registry_with_c99_stdint_types(result)
     fill_registry_with_c99_complex_types(result)
     return DTypeRegistryWrapper(result)

Example #3

File: cuda.py Project: connorjward/loopy

    def get_dtype_registry(self):
        from loopy.target.c.compyte.dtypes import (
            DTypeRegistry, fill_registry_with_opencl_c_types)

        result = DTypeRegistry()
        fill_registry_with_opencl_c_types(result)

        # no complex number support--needs PyOpenCLTarget

        _register_vector_types(result)

        return result

Example #4

File: opencl.py Project: ml-lab/loopy

    def get_dtype_registry(self):
        from loopy.target.c.compyte.dtypes import (
            DTypeRegistry, fill_registry_with_opencl_c_types)

        result = DTypeRegistry()
        fill_registry_with_opencl_c_types(result)

        # no complex number support--needs PyOpenCLTarget

        _register_vector_types(result)

        if self.atomics_flavor == "cl1":
            return DTypeRegistryWrapperWithCL1Atomics(result)
        else:
            raise NotImplementedError("atomics flavor: %s" %
                                      self.atomics_flavor)