`core.resharding.nvshmem_copy_service.memory.double_buffer_manager`#

Double buffer management for NVSHMEM symmetric memory.

Manages send and receive buffers with double-buffering for pipelined communication.

Module Contents#

Manages double-buffered NVSHMEM symmetric buffers for send/receive operations.

class core.resharding.nvshmem_copy_service.memory.double_buffer_manager.DoubleBufferManager(slot_size: int = MAX_SEGMENT_SIZE)#

Manages double-buffered NVSHMEM symmetric buffers for send/receive operations.

Initialization

Initialize buffer manager.

Parameters:: slot_size – Size of each buffer slot in bytes (default: 256MB)

allocate() → None#: Allocate NVSHMEM symmetric buffers for double-buffering.

get_send_slot(iteration: int)#

Get send buffer for given iteration.

get_recv_slot(iteration: int)#

Get receive buffer for given iteration.