cuda.tile.bitwise_xor#
- cuda.tile.bitwise_xor(x, y, /)#
Elementwise bitwise_xor on two tiles.
Can also use builtin operation x ^ y.
The shape of x and y will be broadcasted and dtype promoted to common dtype.
- Return type:
Examples
tx = ct.full((4,), 7, 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,), 7, 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
[4, 4, 4, 4]