triton.language.tensor_descriptor¶
- class triton.language.tensor_descriptor(self, handle, shape: List[tensor], strides: List[tensor], block_type: block_type)¶
A descriptor representing a tensor in global memory.
- __init__(self, handle, shape: List[tensor], strides: List[tensor], block_type: block_type)¶
Not called by user code.
Methods
__init__
(self, handle, shape, strides, ...)Not called by user code.
atomic_add
(self, offsets, value[, _semantic])atomic_and
(self, offsets, value[, _semantic])atomic_max
(self, offsets, value[, _semantic])atomic_min
(self, offsets, value[, _semantic])atomic_or
(self, offsets, value[, _semantic])atomic_xor
(self, offsets, value[, _semantic])gather
(self, *args[, _semantic])Gather multiple descriptors worth of data
load
(self, offsets[, _semantic])Load a block from the descriptor starting at the given element offsets.
scatter
(self, value, *args[, _semantic])Scatter multiple descriptors worth of data
store
(self, offsets, value[, _semantic])Store a block from the descriptor starting at the given element offsets.
Attributes
block_shape
block_type
dtype
type