Fix: preserve early dispatch for late-wired consumers

[ChaoWao]

Summary

• Serialize the dispatch-propagation claim with the producer fanout snapshot so consumer wiring can distinguish included edges from late edges.
• Seed fully published producers wired after one-shot propagation, and enqueue whichever contribution completes dispatch fanin.
• Add sequential and forced-interleaving regression coverage and document the lock-order invariant.
• Validate 34 wiring tests, 60 non-hardware C++ tests, and the A2A3 sync-start early-dispatch scene on simulator and device.

Related Issues

• Supports perf(qwen3-14b decode): fold RoPE into the CANN FAI attention extern pypto-lib#796.

[coderabbitai]

Important

Review skipped

Auto incremental reviews are disabled on this repository.

Please check the settings in the CodeRabbit UI or the .coderabbit.yaml file in this repository. To trigger a single review, invoke the @coderabbitai review command.

⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro

Run ID: 9f4f2cad-808d-4fa6-857c-b290db2ce68e

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

Use the checkbox below for a quick retry:

• 🔍 Trigger review

📝 Walkthrough

Walkthrough

Early-dispatch fanin propagation now accounts for producers that already propagated dispatch state. Wiring seeds late consumers accordingly, scheduler enrollment is centralized and serialized, and tests cover late-wiring and concurrent-propagation cases.

Changes

Early dispatch fanin

Layer / File(s)	Summary
Serialize propagation and centralize candidate enrollment src/a2a3/runtime/tensormap_and_ringbuffer/runtime/scheduler/pto_scheduler.h, src/a2a3/runtime/tensormap_and_ringbuffer/docs/RUNTIME_LOGIC.md	Propagation claims and fanout snapshots are serialized under fanout_lock; eligible consumers are staged once and routed to the appropriate early-dispatch queue.
Seed fanin during wiring src/a2a3/runtime/tensormap_and_ringbuffer/runtime/pto_orchestrator.cpp, src/a2a3/runtime/tensormap_and_ringbuffer/docs/RUNTIME_LOGIC.md	Wiring counts completed and already-propagated producers, updates dispatch_fanin, and enqueues consumers when early fanin is complete.
Validate late and concurrent wiring tests/ut/cpp/a2a3/test_wiring.cpp, src/a2a3/runtime/tensormap_and_ringbuffer/runtime/scheduler/scheduler_dispatch.cpp	Tests cover late wiring and concurrent propagation; the early-dispatch comment is updated to match gated pending-slot behavior.

Estimated code review effort: 3 (Moderate) | ~20 minutes

Sequence Diagram(s)

sequenceDiagram
  participant Producer
  participant PTO2OrchestratorState
  participant PTO2SchedulerState
  participant EarlyDispatchQueue
  Producer->>PTO2SchedulerState: Propagate published dispatch fanin
  PTO2SchedulerState->>PTO2SchedulerState: Claim propagation and snapshot fanout
  PTO2OrchestratorState->>PTO2OrchestratorState: Wire consumer and count early producers
  PTO2OrchestratorState->>PTO2SchedulerState: Enqueue eligible consumer when fanin is complete
  PTO2SchedulerState->>EarlyDispatchQueue: Stage consumer once

Loading

Possibly related PRs

• hw-native-sys/simpler#1314: Both changes modify early-dispatch fanin and candidate eligibility handling in pto_scheduler.h.

Poem

I’m a bunny with dispatch in my toes,
Counting fanin wherever it goes.
Late wires join the queue with care,
Locked fanouts keep counts fair.
Propagate once—then hop away! 🐇

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title clearly matches the main change: preserving early dispatch for consumers wired after producer propagation.
Description check	✅ Passed	The description is directly related to the code changes, tests, and documentation updates in this pull request.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands.}

[gemini-code-assist]

Code Review

This pull request refactors and improves the early-dispatch scheduling logic by ensuring that late-wired consumers are correctly enqueued when their producers are already fully published. It introduces a helper function try_enqueue_early_dispatch_candidate to centralize the enqueuing logic, optimizes lock contention with an unlocked precheck on dispatch_propagated, and updates the documentation and unit tests accordingly. There are no review comments, so I have no feedback to provide.

Important

The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.