[Common] Fix fused router for large top-K and expert counts#2821

Merged

tdophung merged 9 commits into

NVIDIA:mainfrom

harryzhou2000:hhanyu/router_fix_p2

Apr 16, 2026

Commits on Apr 16, 2026

fix: enabling fused _router to be able to handle large topk and number of experts
harryzhou2000
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
harryzhou2000
committed
added topk>=16 in tests/pytorch/test_fused_router.py
harryzhou2000
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
harryzhou2000
committed
test_fused_router.py needs to skip topk >= num_experts case
harryzhou2000
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
harryzhou2000
committed
applying suggesions
harryzhou2000
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
harryzhou2000
committed
warning about dtype for tolerance in test_fused_router.py
harryzhou2000
committed