flashinfer.comm.trtllm_lamport_initialize

flashinfer.comm.trtllm_lamport_initialize(buffer_ptr: int, size: int, dtype: dtype) → None

Initialize a single Lamport-style buffer to negative zero.

Parameters:

• buffer_ptr (int) – Device pointer to the buffer.

• size (int) – Number of elements in the buffer.

• dtype (torch.dtype) – Element dtype of the buffer.