triton.experimental.gluon.language.allocate_shared_memory
- triton.experimental.gluon.language.allocate_shared_memory(element_ty, shape, layout, value=None, _semantic=None) → shared_memory_descriptor
Allocate shared memory for a tensor with the given element type, shape, and layout.
- Parameters:
element_ty (dtype) – The element data type.
shape (Sequence[int]) – The dimensions of the shared memory.
layout (SharedLayout) – The shared memory layout.
value (tensor, optional) – Initial value to copy into shared memory.
- Returns:
Descriptor for the allocated memory.
- Return type: