Python CudaObject Examples

Example #1

File: _interface.py Project: rgommers/cupy

def _create_dim3(name):
    return Dim3(
        _compile.CudaObject(f'{name}.x', _types.uint32),
        _compile.CudaObject(f'{name}.y', _types.uint32),
        _compile.CudaObject(f'{name}.z', _types.uint32),
    )

Example #2

File: _interface.py Project: jbdoderlein/Construction_NN

 def __init__(self, name):
     self.x = _compile.CudaObject(f'{name}.x', _types.uint32)
     self.y = _compile.CudaObject(f'{name}.y', _types.uint32)
     self.z = _compile.CudaObject(f'{name}.z', _types.uint32)
     self.__doc__ = f"""dim3 {name}