[NVIDIA] Support flashinfer Mega Moe (#31470)
Co-authored-by: djns99 <40156487+djns99@users.noreply.github.com> Co-authored-by: 云挚 <ningyunxiao.nyx@antgroup.com> Co-authored-by: Yangmin Li <yangminl@nvidia.com> Co-authored-by: Po-Han Huang (NVIDIA) <53919306+nvpohanh@users.noreply.github.com>
This commit is contained in:
co-authored by
djns99
云挚
Yangmin Li
Po-Han Huang
parent
c0b790cf7f
commit
1b77f498a0
@@ -55,6 +55,7 @@ dependencies = [
|
||||
"nvidia-cutlass-dsl[cu13]==4.6.2",
|
||||
"nvidia-mathdx==25.6.0",
|
||||
"nvidia-ml-py",
|
||||
"nvshmem4py-cu13",
|
||||
"openai==2.6.1",
|
||||
"openai-harmony==0.0.4",
|
||||
"orjson",
|
||||
|
||||
Reference in New Issue
Block a user