warp.tile_zeros#

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

  • Kernel

Allocate a tile of zero-initialized items.

Parameters:

  • shape – Shape of the output tile

  • dtype – Data type of output tile’s elements (default float)

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

Returns:

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

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

  • Kernel

Allocate a tile of zero-initialized items.