Skip to content

refactor(ep/v2): make TokOffExt public for out-of-op callers - #708

Merged
jhchouuu merged 1 commit into
ROCm:mainfrom
jhchouuu:jhchou/megamoe-tokoff-public-export
Sep 24, 2026
Merged

jhchouuu merged 1 commit into
ROCm:mainfrom
jhchouuu:jhchou/megamoe-tokoff-public-export

Conversation

@jhchouuu

Copy link
Copy Markdown
Collaborator

Rename _TokOffExt to TokOffExt so callers outside the op layer can use it.

aiter's MegaMoE drives EpDispatchPlan directly and needs this buffer to bind EpArgs.tokOffPeers; a public name lets it import from the module instead of a private one. No behaviour change.

🤖 Generated with Claude Code

The tokoff-ext slot allocator is built by EpDispatchCombineOpHip, but
aiter's MegaMoE drives EpDispatchPlan directly and needs the same buffer
to bind EpArgs.tokOffPeers. Drop the underscore so that path imports it
from this module the way it already imports EpDispatchPlan, rather than
reaching for a private name.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
@jhchouuu
jhchouuu merged commit 6cd0ca4 into ROCm:main Sep 24, 2026
5 of 6 checks passed
@jhchouuu
jhchouuu deleted the jhchou/megamoe-tokoff-public-export branch September 24, 2026 05:12
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