cuda.tile.bitwise_or#

cuda.tile.bitwise_or(x, y, /)#

Elementwise bitwise_or on two tiles.

Can also use builtin operation x | y.

Parameters:

x (Tile) – LHS tile.
y (Tile) – RHS tile.

The shape of x and y will be broadcasted and dtype promoted to common dtype.

Return type:: Tile

Examples

SnippetComplete Example

tx = ct.full((4,), 5, dtype=ct.int32)
ty = ct.full((4,), 3, dtype=ct.int32)
print(tx | ty)

import cuda.tile as ct
import torch

@ct.kernel
def kernel():
    tx = ct.full((4,), 5, dtype=ct.int32)
    ty = ct.full((4,), 3, dtype=ct.int32)
    print(tx | ty)


torch.cuda.init()
ct.launch(torch.cuda.current_stream(), (1,), kernel, ())
torch.cuda.synchronize()

Output

[7, 7, 7, 7]