free_symmetric_memory#

nvmath.distributed.free_symmetric_memory(*tensors) None[source]#

Frees tensors’ data buffer where the buffer was allocated on the symmetric heap. Note that this is only meant to be called on tensors returned by allocate_symmetric_memory().

This is a collective operation and must be called by all processes, with tensors in the same order.