Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
3628bcaa
Unverified
Commit
3628bcaa
authored
Dec 05, 2025
by
Zhiwei
Committed by
GitHub
Dec 05, 2025
Browse files
[ROCm][MXFP4] Infer w4a4 quant method in rocm aiter fused moe (#29775)
Signed-off-by:
ZhiweiYan-96
<
zhiwei.yan@amd.com
>
parent
b73b158a
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
2 deletions
+6
-2
vllm/model_executor/layers/fused_moe/config.py
vllm/model_executor/layers/fused_moe/config.py
+4
-0
vllm/model_executor/layers/fused_moe/rocm_aiter_fused_moe.py
vllm/model_executor/layers/fused_moe/rocm_aiter_fused_moe.py
+2
-2
No files found.
vllm/model_executor/layers/fused_moe/config.py
View file @
3628bcaa
...
...
@@ -345,6 +345,10 @@ class FusedMoEQuantConfig:
def
use_mxfp4_w4a16
(
self
)
->
bool
:
return
self
.
_a1
.
dtype
is
None
and
self
.
_w1
.
dtype
==
"mxfp4"
@
property
def
use_mxfp4_w4a4
(
self
)
->
bool
:
return
self
.
_a1
.
dtype
==
"mxfp4"
and
self
.
_w1
.
dtype
==
"mxfp4"
@
property
def
use_nvfp4_w4a4
(
self
)
->
bool
:
return
self
.
quant_dtype
==
"nvfp4"
...
...
vllm/model_executor/layers/fused_moe/rocm_aiter_fused_moe.py
View file @
3628bcaa
...
...
@@ -221,8 +221,8 @@ def rocm_aiter_fused_experts(
else
:
quant_method
=
QuantMethod
.
NO
.
value
# quark moe for mxfp4 w_dtype
if
quant_config
.
use_mxfp4_w4a
16
:
# quark moe for mxfp4 w_dtype
mxfp4 a_dtype
if
quant_config
.
use_mxfp4_w4a
4
:
quant_method
=
QuantMethod
.
BLOCK_1X32
.
value
# w8a8 block-scaled
if
quant_config
.
block_shape
is
not
None
and
quant_config
.
use_fp8_w8a8
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment