[AMD][EPLB] Enable EPLB for Quark OCP MXFP4 MoE (#47220)

Signed-off-by: okorzh-amd <okorzh-amd@users.noreply.github.com>
Co-authored-by: okorzh-amd <okorzh-amd@users.noreply.github.com>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Co-authored-by: Andreas Karatzas <akaratza@amd.com>
This commit is contained in:
Oxana Korzh
2026-07-03 14:41:52 -05:00
committed by GitHub
co-authored by okorzh-amd Claude Opus 4.8 Andreas Karatzas
parent f006e5a24c
commit 576bf75d0e
@@ -1322,6 +1322,11 @@ class QuarkOCP_MX_MoEMethod(QuarkMoEMethod):
gemm1_clamp_limit=getattr(layer, "swiglu_limit", None),
)
@property
def supports_eplb(self) -> bool:
# AITER shuffle keeps expert dim outermost, so EPLB row moves are layout-safe.
return True
@property
def is_monolithic(self) -> bool:
if self.moe_kernel is not None: