Skip to content

[Graph Optimization] Wrap m_grouped_gemm_fp8_fp8_bf16_nt_contiguous as custom pyop#5847

Merged
EmmonsCurse merged 1 commit intoPaddlePaddle:developfrom
cattidea:custom_python_op/m_grouped_gemm_fp8_fp8_bf16_nt_contiguous_custom
Jan 7, 2026
Merged

[Graph Optimization] Wrap m_grouped_gemm_fp8_fp8_bf16_nt_contiguous as custom pyop#5847
EmmonsCurse merged 1 commit intoPaddlePaddle:developfrom
cattidea:custom_python_op/m_grouped_gemm_fp8_fp8_bf16_nt_contiguous_custom

Conversation

@DrRyanHuang
Copy link
Copy Markdown
Collaborator

@DrRyanHuang DrRyanHuang commented Dec 31, 2025

Motivation

支持 ERNIE 系列模型转静

Modifications

将 jit 算子m_grouped_gemm_fp8_fp8_bf16_nt_contiguous wrap 为 Python 自定义算子

Usage or Command

NO NEED

Accuracy Tests

NO NEED

Checklist

  • Add at least a tag in the PR title.
    • Tag list: [[FDConfig],[APIServer],[Engine], [Scheduler], [PD Disaggregation], [Executor], [Graph Optimization], [Speculative Decoding], [RL], [Models], [Quantization], [Loader], [OP], [KVCache], [DataProcessor], [BugFix], [Docs], [CI], [Optimization], [Feature], [Benchmark], [Others], [XPU], [HPU], [GCU], [DCU], [Iluvatar], [Metax]]
    • You can add new tags based on the PR content, but the semantics must be clear.
  • Format your code, run pre-commit before commit.
  • Add unit tests. Please write the reason in this PR if no unit tests.
  • Provide accuracy results.
  • If the current PR is submitting to the release branch, make sure the PR has been submitted to the develop branch, then cherry-pick it to the release branch with the [Cherry-Pick] PR tag.

@paddle-bot
Copy link
Copy Markdown

paddle-bot bot commented Dec 31, 2025

Thanks for your contribution!

@codecov-commenter
Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 0% with 18 lines in your changes missing coverage. Please review.
⚠️ Please upload report for BASE (develop@9286403). Learn more about missing BASE report.

Files with missing lines Patch % Lines
..._executor/layers/moe/fused_moe_deepgemm_backend.py 0.00% 18 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             develop    #5847   +/-   ##
==========================================
  Coverage           ?   66.83%           
==========================================
  Files              ?      347           
  Lines              ?    44363           
  Branches           ?     6816           
==========================================
  Hits               ?    29649           
  Misses             ?    12528           
  Partials           ?     2186           
Flag Coverage Δ
GPU 66.83% <0.00%> (?)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

Copy link
Copy Markdown
Collaborator

@gongshaotian gongshaotian left a comment

Choose a reason for hiding this comment

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

LGTM

@EmmonsCurse EmmonsCurse merged commit 3e74bac into PaddlePaddle:develop Jan 7, 2026
16 of 20 checks passed
@DrRyanHuang DrRyanHuang deleted the custom_python_op/m_grouped_gemm_fp8_fp8_bf16_nt_contiguous_custom branch January 7, 2026 08:21
chang-wenbin pushed a commit to chang-wenbin/FastDeploy that referenced this pull request Mar 2, 2026
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.

4 participants