Skip to content

[Common] Add dense router output for fused router#3129

Open
harryzhou2000 wants to merge 8 commits into
NVIDIA:mainfrom
harryzhou2000:hhanyu/router_dense_route
Open

[Common] Add dense router output for fused router#3129
harryzhou2000 wants to merge 8 commits into
NVIDIA:mainfrom
harryzhou2000:hhanyu/router_dense_route

[PyTorch] Preserve router leading dimensions

49c6553
Select commit
Loading
Failed to load commit list.