Skip to content

Implement pipelined H2dWrite (op=7) via BlockTransport - #452

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_953988360
Open

Implement pipelined H2dWrite (op=7) via BlockTransport#452
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_953988360

Conversation

@copybara-service

Copy link
Copy Markdown

Implement pipelined H2dWrite (op=7) via BlockTransport

This CL resolves the remaining host block ID aliasing bug by enabling
in-band sender-driven H2D pipelines over BlockTransport op=7.

  • Extending BlockTransport to serialize dst_device_block_ids
  • Pipelining OnLayerReceived to trigger H2D DMA securely
  • Retaining fallback shadowing in KVCacheManagerWithTransfer
  • Strictly blocking final network ACK to guarantee single-call HBM completion

This CL resolves the remaining host block ID aliasing bug by enabling
in-band sender-driven H2D pipelines over BlockTransport op=7.
- Extending BlockTransport to serialize dst_device_block_ids
- Pipelining OnLayerReceived to trigger H2D DMA securely
- Retaining fallback shadowing in KVCacheManagerWithTransfer
- Strictly blocking final network ACK to guarantee single-call HBM completion

PiperOrigin-RevId: 953988360
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant