cuda.tile.full

cuda.tile.full(shape, fill_value, dtype)

Creates a tile filled with given value.

Parameters:

• shape (tuple[const int,...]) – The shape of the tile.

• fill_value (int | float | bool]) – Value for the tile.

• dtype (DType) – The Data type of the tile.

Return type:

Tile

Examples

SnippetComplete Example

tile = ct.full((2, 2), 3, dtype=ct.int32)
print(tile)

Copy

import cuda.tile as ct
import torch

@ct.kernel
def kernel():
    tile = ct.full((2, 2), 3, dtype=ct.int32)
    print(tile)


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

Output

[[3, 3], [3, 3]]