For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo function
tcgen05_alloc
def tcgen05_alloc[cta_group: Int32](ptr_tmem_addr: Pointer[UInt32, address_space=AddressSpace.SHARED, _safe=True], num_cols: UInt32)
Allocates tensor memory for use with tcgen05 instructions.
Note: This function is only available on NVIDIA Blackwell GPUs (SM 100+).
Parameters:
- cta_group (
Int32): The cooperative thread array (CTA) group ID.
Args:
- ptr_tmem_addr (
Pointer[UInt32, address_space=AddressSpace.SHARED, _safe=True]): Shared memory pointer to hold tensor memory address. - num_cols (
UInt32): The number of columns to allocate.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!