Skip to main content
Log in

Mojo function

tcgen05_dealloc

tcgen05_dealloc[cta_group: SIMD[int32, 1]](tmem_addr: SIMD[uint32, 1], num_cols: SIMD[uint32, 1])

Deallocates tensor memory allocated by tcgen05_alloc().

This function deallocates tensor memory that was previously allocated using tcgen05_alloc(). The deallocation must be performed by the same CTA group that performed the allocation.

Parameters:

  • cta_group (SIMD[int32, 1]): The cooperative thread array (CTA) group ID.

Args:

  • tmem_addr (SIMD[uint32, 1]): Address of the tensor memory to deallocate.
  • num_cols (SIMD[uint32, 1]): Number of columns in the tensor memory.

Was this page helpful?