All pull requests

Pull requests

• Export Peregrine transport tier-1 metrics (1/N)

  #1092·copybara-service[bot] opened Sep 29, 2026Bot·

  ·
  1

• Avoid decode accept() busy-spin and prefill socket worker freeze on accept failure.

  #1091·copybara-service[bot] opened Sep 28, 2026Bot·

  ·
  1

• [TPU/Raiden/Hardening] Bound data-plane connect() with a timeout

  #1090·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• [tpu_raiden] Refactor block transport to Peregrine-like post/poll API (15/N)

  #1089·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• Record p2p_transfer_time_ms on pull_response, attach local_rank label for parallel TP worker isolation, and clamp overlapping stream durations in socket_transport_adapter.cc.

  #1088·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• [tpu_raiden] Refactor block transport to Peregrine-like post/poll API (14/N)

  #1086·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• Prototype: opt-in flags to drop libbpf, libacl and zeromq from GGC builds

  #1083·copybara-service[bot] opened Sep 28, 2026Bot·

  ·
  1

• Fixed weights sync tree broadcast checksum mismatch across multi-stage pipelines.

  #1082·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• Add CPU repro test for tree-broadcast seed/relay completion defects.

  #1081·copybara-service[bot] opened Sep 28, 2026Bot·

  ·

• Add a TPU7x test that reproduces compute / DMA overlal raw DMA kernel stalls.

  #1080·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Pace tpu_raiden torch raw DMA so kernels don't wait for whole copies.

  #1079·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Bump tpu_raiden JAX build to 0.11.2 and support runtime JAX >= 0.11.0

  #1076·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Add PyTorch unit test suites (tpu_sync/frameworks/torch:all and tpu_sync/api/torch:all) to TAP blueprint test_patterns.

  #1071·copybara-service[bot] opened Sep 27, 2026Bot·

  ·
  1

• Instantiate tpu_raiden::jax::WeightSynchronizer in JAX weight synchronizer FFI modules.

  #1070·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Add bounded host ring-buffer pools and DP sub-sharding with post-H2D ICI all-gather.

  #1069·copybara-service[bot] opened Sep 27, 2026Bot·

  ·

• Add an opt-in cache of shard files confirmed on storage to

  #1068·copybara-service[bot] opened Sep 26, 2026Bot·

  ·

• Add offline symbolic and parallel weight-sync plan computation.

  #1066·copybara-service[bot] opened Sep 26, 2026Bot·

  ·

• Reshard KV pools between pipeline-parallel engines

  #1064·copybara-service[bot] opened Sep 26, 2026Bot·

  ·

• - Replace the single-slot ShardScratchpad in WeightSynchronizerBase with a 2-slot (kNumScratchpadSlots = 2) double-buffered scratchpad per shard (ScratchpadSlot slots[2]), allowing CPU tiling/detiling on one slot to overlap with in-flight PCIe H2D/D2H DMA (CopyRawHostToDevice / CopyRawDeviceToHost) on the other slot without serializing on in_flight_future.Await().

  #1063·copybara-service[bot] opened Sep 26, 2026Bot·

  ·

• Reshard KV pools between pipeline-parallel engines

  #1062·yinlin09 opened Sep 25, 2026Collaborator·

  ·

• [tpu_sync] Support multi-backend Load() with secondary storage suffix in KVCacheStore.

  #1061·copybara-service[bot] opened Sep 25, 2026Bot·

  ·

• [tpu_sync] Replace copied Peregrine code with peregrine v0.1.2 dependency

  #1054·copybara-service[bot] opened Sep 25, 2026Bot·

  ·
  1

• [tpu_raiden] Refactor block transport to Peregrine-like post/poll API (13/N)

  #1050·copybara-service[bot] opened Sep 25, 2026Bot·

  ·

• Allow configuring fault injection rules via a plain text file. When the file is absent - no rules are installed.

  #1045·copybara-service[bot] opened Sep 25, 2026Bot·

  ·
  1

• [TPU Raiden] Fix [Failure/ReceiveDisconnect]: propagate broken inbound streams to fail active receive sessions and release staging slots immediately.

  #1041·copybara-service[bot] opened Sep 24, 2026Bot·

  ·
  1