-
Notifications
You must be signed in to change notification settings - Fork 693
[Feature] Support Ernie FP8 on sm100 #5593
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Feature] Support Ernie FP8 on sm100 #5593
Conversation
|
Thanks for your contribution! |
|
K11OntheBoat seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account. You have signed the CLA already but the status is still pending? Let us recheck it. |
5d43f4e to
e608526
Compare
6079bc8 to
bb9c247
Compare
bb9c247 to
751a688
Compare
f671841 to
b4431a0
Compare
Codecov Report❌ Patch coverage is Additional details and impacted files@@ Coverage Diff @@
## develop #5593 +/- ##
==========================================
Coverage ? 66.46%
==========================================
Files ? 385
Lines ? 50892
Branches ? 7944
==========================================
Hits ? 33827
Misses ? 14578
Partials ? 2487
Flags with carried forward coverage won't be shown. Click here to find out more. ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
faf3f21 to
9377f0a
Compare
9377f0a to
e2c0146
Compare
e2c0146 to
60cfd25
Compare
3b0e282 to
02906fe
Compare
| def _get_mn_major_tma_aligned_packed_ue8m0_tensor_torch_impl( | ||
| x: paddle.Tensor, | ||
| ): | ||
| """将FP32张量转换为TMA对齐的packed UE8M0格式张量""" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
是否考虑换成英文注释
| from fastdeploy.model_executor.ops.gpu.deep_gemm import ceil_div | ||
| try: | ||
| from deep_gemm import ceil_div | ||
| except ModuleNotFoundError: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
最好打印出warning吧
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
好的,这两个问题在后续PR一并修复
This reverts commit eb80724.
Motivation
This PR is going to support Blockwise-FP8 inference on SM100.
Modifications
paddle.incubate.nn.functional.fp8_quant_blockwiseinstead ofper_token_quant/per_token_quant_padding. Add support ue8m0 formasked_per_token_quant, using in EP Decoding phase.Usage or Command
requires paddlepaddle develop after(include) 20260119
For DeepGEMM installtion: https://github.com/PFCCLab/DeepGEMM
For DeepEP installtion: https://github.com/PFCCLab/DeepEP
Accuracy Tests
Checklist
[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]]pre-commitbefore commit.releasebranch, make sure the PR has been submitted to thedevelopbranch, then cherry-pick it to thereleasebranch with the[Cherry-Pick]PR tag.