[MoE Refactor] Integrate Naive Prepare Finalize into MK#32567

Merged

robertgshaw2-redhat merged 109 commits intomainfrom

naive-pf-separation

Jan 27, 2026

Commits on Jan 19, 2026

a2a
Robert Shaw
committed
base device communicator
Robert Shaw
committed
rest of the device communicators
Robert Shaw
committed
add moe prepare and finalize naive ep
Robert Shaw
committed
get naive dispatch/combine into a good spot
Robert Shaw
committed
update maybe_make_prepare_finalize and select_gemm_impl
Robert Shaw
committed
update make_nvfp4 oracle
Robert Shaw
committed
make the prepare finalize in the oracle
Robert Shaw
committed
remove some fi functions
Robert Shaw
committed
things are 'working' end to end
Robert Shaw
committed
fix topk indices dtype
Robert Shaw
committed

Commits on Jan 21, 2026

Support nccl fp8 communication (#32760)
amirkl94
authored
merge
Robert Shaw
committed
Merge branch 'naive-pf-separation' of https://github.com/vllm-project/vllm into naive-pf-separation
Robert Shaw
committed
pre-comimt remediation
Robert Shaw
committed
pre-comimt remediation
Robert Shaw
committed
fun with pre-commit
Robert Shaw
committed
remove FlashInferAG
Robert Shaw
committed
updated
Robert Shaw
committed
small nit
Robert Shaw
committed
move mkm
Robert Shaw
committed
rename file
Robert Shaw
committed
support flashinfer cutlass fp8 with DeepEPHT
Robert Shaw
committed
make deepepht prepare-finalize work with cutlass fp8 block
Robert Shaw
committed
stash changes to make nvfp4 unfused
Robert Shaw
committed
do the fused swizzling
Robert Shaw
committed
stash
Robert Shaw
committed
updated to remove the explicit gather call
Robert Shaw
committed
make mnnvl work properly
Robert Shaw
committed
make mnnvl work properly
Robert Shaw
committed
remove bad import
Robert Shaw
committed
add compatibility for flashinfer mnnvl
Robert Shaw
committed
add todo
Robert Shaw
committed

Commits on Jan 23, 2026

merge main
Robert Shaw
committed
pre-commit
Robert Shaw
committed
update type signatures
Robert Shaw
committed
update type signatures
Robert Shaw
committed

Commits on Jan 24, 2026

make pre-commit happy
Robert Shaw
committed
get tests passing
Robert Shaw
committed
self.kernel -> self.mk_moe; move shared items up to the base class
Robert Shaw
committed
self.kernel -> self.mk_moe; move shared items up to the base class
Robert Shaw
committed
nit
Robert Shaw
committed
updated
Robert Shaw
committed
reduce LOC change
Robert Shaw
committed
update comments
Robert Shaw
committed
revert removal of extra tesnors from dispatch/combine
Robert Shaw
committed
reduce loc change
Robert Shaw
committed
remove extratensors for naive A2a
Robert Shaw
committed
moe_mk -> mk_moe
Robert Shaw
committed
moe_mk -> mk_moe
Robert Shaw
committed
fix flashinfer 0.5.3 -> 0.6.1 update
Robert Shaw
committed
remove support for defer input quant to batched P/Fes
Robert Shaw
committed
Merge remote-tracking branch 'origin/main' into naive-pf-separation
Robert Shaw
committed
fixed bug with ag/rs and nvfp4 quantization related the the swizzling
Robert Shaw
committed
updated
Robert Shaw
committed
fix up cutlass + pplx
Robert Shaw
committed
fix up pre-commit
Robert Shaw
committed
fix up pre-commit
Robert Shaw
committed
make pre-commit support other routing strategy
Robert Shaw
committed
remove debug cruft
Robert Shaw
committed
reduce LOC change
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed

Commits on Jan 25, 2026

fix handling of monolithic kernels
Robert Shaw
committed
Merge remote-tracking branch 'origin/main' into naive-pf-separation
Robert Shaw
committed
fix AG/RS dtype
Robert Shaw
committed
fix incorrect arg in test
Robert Shaw
committed
fix missing args
Robert Shaw
committed
fix mk
Robert Shaw
committed
fix pre-commit
Robert Shaw
committed
fix docs build
Robert Shaw
committed
confirmed shared experts working
Robert Shaw
committed
update, remove to cuda
Robert Shaw
committed
verify
Robert Shaw
committed
fix DP/TP deployment
Robert Shaw
committed

Commits on Jan 26, 2026

update defer input quant
Robert Shaw
committed
merge main
Robert Shaw
committed
remove empty file
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
reduce LOC changed
Robert Shaw
committed
update comment
Robert Shaw
committed
Merge branch 'main' into naive-pf-separation
robertgshaw2-redhat
authored
Merge branch 'main' into naive-pf-separation
robertgshaw2-redhat
authored
Merge branch 'main' into naive-pf-separation
robertgshaw2-redhat
authored
fix docstring
Robert Shaw
committed
add comemnta bout commBackned
Robert Shaw
committed
add comemnta bout commBackned
Robert Shaw
committed
Merge branch 'main' into naive-pf-separation
robertgshaw2-redhat
authored