warp.tile_arange#

warp.tile_arange(
*args: Scalar,
dtype: Scalar,
storage: str,
) → Tile[Scalar, tuple[int]]#

  • Kernel

Generate a tile of linearly spaced elements.

param args:

Variable-length positional arguments, interpreted as:

  • (stop,): Generates values from 0 to stop - 1

  • (start, stop): Generates values from start to stop - 1

  • (start, stop, step): Generates values from start to stop - 1 with a step size

param dtype:

Data type of output tile’s elements (optional, default: float)

param storage:

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

returns:

A tile with shape=(n) with linearly spaced elements of specified data type