Unverified Commit b7e5a588 authored by Robert Shaw's avatar Robert Shaw Committed by GitHub
Browse files

[Bugfix] Fix DP/EP Shared Expert With Monolithic Kernels (#36061)


Signed-off-by: default avatarRobert Shaw <robshaw@redhat.com>
Co-authored-by: default avatarRobert Shaw <robshaw@redhat.com>
parent 822e250a
......@@ -567,7 +567,7 @@ def make_fp8_moe_kernel(
experts,
shared_experts=(
shared_experts
if moe_config.moe_parallel_config.use_all2all_kernels
if moe_config.moe_parallel_config.use_deepep_ll_kernels
else None
),
moe_parallel_config=moe_config.moe_parallel_config,
......
......@@ -433,7 +433,7 @@ def make_nvfp4_moe_kernel(
experts,
shared_experts=(
shared_experts
if moe_config.moe_parallel_config.use_all2all_kernels
if moe_config.moe_parallel_config.use_deepep_ll_kernels
else None
),
moe_parallel_config=moe_config.moe_parallel_config,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment