Skip to content

Register Triton implementation as a custom op to fix the Mixer2 gated RMS norm lowering issue.#14

Open
chaojun-zhang wants to merge 1 commit into
wxsIcey:wxs/vllm-ir-mixer2-gated-rms-normfrom
chaojun-zhang:vllm-ir-mixer2-gated-rms-norm
Open

Register Triton implementation as a custom op to fix the Mixer2 gated RMS norm lowering issue.#14
chaojun-zhang wants to merge 1 commit into
wxsIcey:wxs/vllm-ir-mixer2-gated-rms-normfrom
chaojun-zhang:vllm-ir-mixer2-gated-rms-norm

Conversation

@chaojun-zhang

@chaojun-zhang chaojun-zhang commented Apr 22, 2026

Copy link
Copy Markdown

Purpose

  1. Add a test for Mixer2 gated RMS norm lowering
  2. Register Triton implementation as a custom op to fix the lowering issue.

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.

Signed-off-by: Chaojun Zhang <chaojun.zhang@intel.com>
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