warp.tile_extract#
- warp.tile_extract(a: Tile[Any, tuple[int]], i: int32) Any#
Kernel
Differentiable
Extract a single element from the tile.
This function will extract an element from the tile and broadcast its value to all threads in the block.
Note that this may incur additional synchronization if the source tile is a register tile.
- Parameters:
a – Tile to extract the element from
i – Coordinate of element on first dimension
- Returns:
The value of the element at the specified tile location with the same data type as the input tile.
- warp.tile_extract( ) Any
Kernel
Differentiable
Extract a single element from the tile.
This function will extract an element from the tile and broadcast its value to all threads in the block.
Note that this may incur additional synchronization if the source tile is a register tile.
- Parameters:
a – Tile to extract the element from
i – Coordinate of element on first dimension
j – Coordinate of element on the second dimension, or vector index
- Returns:
The value of the element at the specified tile location with the same data type as the input tile.
- warp.tile_extract( ) Any
Kernel
Differentiable
Extract a single element from the tile.
This function will extract an element from the tile and broadcast its value to all threads in the block.
Note that this may incur additional synchronization if the source tile is a register tile.
- Parameters:
a – Tile to extract the element from
i – Coordinate of element on first dimension
j – Coordinate of element on the second dimension, or first matrix index
k – Coordinate of element on the third dimension, or vector index, or second matrix index
- Returns:
The value of the element at the specified tile location with the same data type as the input tile.
- warp.tile_extract( ) Any
Kernel
Differentiable
Extract a single element from the tile.
This function will extract an element from the tile and broadcast its value to all threads in the block.
Note that this may incur additional synchronization if the source tile is a register tile.
- Parameters:
a – Tile to extract the element from
i – Coordinate of element on first dimension
j – Coordinate of element on the second dimension
k – Coordinate of element on the third dimension, or first matrix index
l – Coordinate of element on the fourth dimension, or vector index, or second matrix index
- Returns:
The value of the element at the specified tile location, with the same data type as the input tile.
- warp.tile_extract( ) Any
Kernel
Differentiable
Extract a single element from the tile.
This function will extract an element from the tile and broadcast its value to all threads in the block.
Note that this may incur additional synchronization if the source tile is a register tile.
- Parameters:
a – Tile to extract the element from
i – Coordinate of element on first dimension
j – Coordinate of element on the second dimension
k – Coordinate of element on the third dimension
l – Coordinate of element on the fourth dimension, or first matrix index
m – Vector index, or second matrix index
- Returns:
The value of the element at the specified tile location, with the same data type as the input tile.
- warp.tile_extract(
- a: Tile[Any, tuple[int, int, int, int]],
- i: int32,
- j: int32,
- k: int32,
- l: int32,
- m: int32,
- n: int32,
Kernel
Differentiable
Extract a single element from the tile.
This function will extract an element from the tile and broadcast its value to all threads in the block.
Note that this may incur additional synchronization if the source tile is a register tile.
- Parameters:
a – Tile to extract the element from
i – Coordinate of element on first dimension
j – Coordinate of element on the second dimension
k – Coordinate of element on the third dimension
l – Coordinate of element on the fourth dimension
m – Vector index, or first matrix index
n – Second matrix index
- Returns:
The value of the element at the specified tile location, with the same data type as the input tile.