Mojo function

tcgen05_dealloc

tcgen05_dealloc[cta_group: SIMD[int32, 1]](mut tmem: TensorMemory)

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.

Note: This function is only available on NVIDIA Blackwell GPUs (SM 100+).

Parameters:

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

Args:

• ​tmem (TensorMemory): TensorMemory struct to hold the allocation address and number of columns.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit