warp.tile_ones#

warp.tile_ones(
shape: tuple[int, ...],
dtype: Any,
storage: str,
) → Tile[Any, tuple[int, ...]]#

  • Kernel

Allocate a tile of one-initialized items.

Parameters:

  • shape – Shape of the output tile

  • dtype – Data type of output tile’s elements

  • storage – The storage location for the tile: "register" for registers (default) or "shared" for shared memory.

Returns:

A one-initialized tile with shape and data type as specified.

warp.tile_ones(
shape: int32,
dtype: Any,
storage: str,
) → Tile[Any, tuple[int, ...]]

  • Kernel

Allocate a tile of one-initialized items.