Skip to content

Pull requests: google/tpu-raiden

New pull request New
22 Open 145 Closed
22 Open 145 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add MT contention benchmark for Raiden
#166 opened Jun 22, 2026 by copybara-service Bot Loading…
Use TF_ASSIGN_OR_RETURN in WriteBlockDirectInternal
#165 opened Jun 22, 2026 by copybara-service Bot Loading…
Rename LogicalBlockManager to HostBlockManager.
#164 opened Jun 21, 2026 by copybara-service Bot Loading…
Refactor TPU Raiden core to disallow C++ exceptions.
#163 opened Jun 20, 2026 by copybara-service Bot Loading…
Allocate host staging slot for inbound blocks in StartRead
#162 opened Jun 20, 2026 by copybara-service Bot Loading…
Handle benign errno in RawBufferTransport::ConnectionWorker poll
#161 opened Jun 20, 2026 by copybara-service Bot Loading…
Assert raw-transfer layout assumptions; use base storage buffer
#160 opened Jun 20, 2026 by copybara-service Bot Loading…
Retry WriteBlockDirect on stale-connection failures
#159 opened Jun 20, 2026 by copybara-service Bot Loading…
Reclaim stale recv entries on timeout in CompleteReadRaw
#158 opened Jun 20, 2026 by copybara-service Bot Loading…
Fix multi-process DmaMap out-of-bounds abort and refactor host memory allocator fallback
#157 opened Jun 20, 2026 by copybara-service Bot Loading…
Implement multi-NIC Key-Value cache transfer routing via explicit shard descriptors (EndpointDescriptor) across JAX and PyTorch Python APIs.
#152 opened Jun 19, 2026 by copybara-service Bot Loading…
Refactor KVCacheManagerWithTransfer control plane to use transport commands API, removing separate control server port and thread.
#149 opened Jun 18, 2026 by copybara-service Bot Loading…
Owning DeviceBufferRef in KV Cache Manager
#146 opened Jun 18, 2026 by copybara-service Bot Loading…
1
Add CPU-only GitHub Actions CI (presubmit + nightly + lint)
#143 opened Jun 18, 2026 by yinlin09 Collaborator Loading…
[Experimental] Raiden H2H setup and benchmark scripts.
#139 opened Jun 17, 2026 by copybara-service Bot Loading…
Implemented resharding KV cache with Raiden controller and block-based push transfer.
#136 opened Jun 17, 2026 by copybara-service Bot Loading…
Implement multi-NIC C++ transport with PCIe-NUMA auto-discovery and FFI registry.
#135 opened Jun 17, 2026 by copybara-service Bot Loading…
Implement multi-NIC transport socket binding and GKE /sys mount to restore dual-NIC performance.
#124 opened Jun 15, 2026 by copybara-service Bot Loading…
Refactor build to link against pre-compiled JAX and PyTorch wheels.
#107 opened Jun 12, 2026 by copybara-service Bot Loading…
[peregrine] extend transport api for posting pipelined requests
#91 opened Jun 11, 2026 by copybara-service Bot Loading…
2
Harden Kokoro presubmit, isolate GCS cache, and add an internal wheel sync script.
#89 opened Jun 11, 2026 by copybara-service Bot Loading…
Use status macro to fix build failure
#56 opened Jun 4, 2026 by copybara-service Bot Loading…
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.