Skip to content

[MoE] Move aiter experts to fused_moe/experts/#41981

Closed
bnellnm wants to merge 6 commits into
vllm-project:mainfrom
neuralmagic:move-aiter-moe-to-experts
Closed

[MoE] Move aiter experts to fused_moe/experts/#41981
bnellnm wants to merge 6 commits into
vllm-project:mainfrom
neuralmagic:move-aiter-moe-to-experts

Conversation

@bnellnm

@bnellnm bnellnm commented May 7, 2026

Copy link
Copy Markdown
Collaborator

Purpose

Rename rocm_aiter_fused_moe.py to experts/rocm_aiter_moe.py and update all references. Also fix stale docs pointer for grouped_topk after its earlier move to router/.

Forked from #40573

cc: @Jackmin801 , @robertgshaw2-redhat

Test Plan

Test Result


Essential Elements of an Effective PR Description Checklist
  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.

Jackmin801 and others added 6 commits April 22, 2026 00:34
Rename rocm_aiter_fused_moe.py to experts/rocm_aiter_moe.py and update
all references. Also fix stale docs pointer for grouped_topk after its
earlier move to router/.

Co-Authored-By: Claude <noreply@anthropic.com>
Signed-off-by: Jackmin801 <ongjackm@gmail.com>
Signed-off-by: Robert Shaw <robertgshaw2@gmail.com>
…xperts

Signed-off-by: Jackmin801 <ongjackm@gmail.com>

# Conflicts:
#	vllm/model_executor/layers/fused_moe/__init__.py
…erts

Signed-off-by: Bill Nell <bnell@redhat.com>

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

@mergify

mergify Bot commented May 7, 2026

Copy link
Copy Markdown
Contributor

Documentation preview: https://vllm--41981.org.readthedocs.build/en/41981/

@mergify mergify Bot added documentation Improvements or additions to documentation rocm Related to AMD ROCm labels May 7, 2026
@github-project-automation github-project-automation Bot moved this to Todo in AMD May 7, 2026

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request refactors the ROCm Aiter MoE implementation by moving components from the top-level fused_moe directory into specialized experts and router subdirectories. The changes include updating import paths across the model executor, quantization layers, tests, and design documentation. I have no feedback to provide.

@bnellnm

bnellnm commented May 8, 2026

Copy link
Copy Markdown
Collaborator Author

Combined into one PR #41979

@bnellnm bnellnm closed this May 8, 2026
@github-project-automation github-project-automation Bot moved this from Todo to Done in AMD May 8, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

documentation Improvements or additions to documentation rocm Related to AMD ROCm

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants