flashinfer.comm.trtllm_lamport_initialize_all

flashinfer.comm.trtllm_lamport_initialize_all(buffer_0_ptr: int, buffer_1_ptr: int, buffer_2_ptr: int, size: int, dtype: dtype) → None

Initialize three Lamport buffers to negative zero.

Parameters:

• buffer_0_ptr (int) – Device pointer to the first buffer.

• buffer_1_ptr (int) – Device pointer to the second buffer.

• buffer_2_ptr (int) – Device pointer to the third buffer.

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

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