Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion packages/paddlefleet_ops/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def get_special_setup_deps():
deps = [
"triton", # for deep_gemm, flashmask
"einops", # for flash-linear-attention
"nvidia-cutlass-dsl[cu13]==4.4.1", # for sonic_moe and flash_attention
"nvidia-cutlass-dsl[cu13]==4.5.0", # for sonic_moe and flash_attention

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

P1 该版本固定会让 FlashMask 的依赖解析失败。当前 third_party/flash-attention/flashmask/setup.py:161 仍声明 nvidia-cutlass-dsl[cu13]>=4.4.1,<=4.4.2,而此处改为 ==4.5.0;构建流程实际执行 FlashMask 的 setup.py,其预安装逻辑会同时处理这两个约束。我用解析器验证两者组合得到 requirements are unsatisfiable,因此 CUDA/NVIDIA 的 ops wheel 构建无法稳定完成。请同步升级 FlashMask 的兼容约束/子模块,或选择同时满足 cuDNN frontend 与 FlashMask 的 CUTLASS DSL 版本,并同步锁文件元数据。

"filelock", # for sonic_moe
"apache-tvm-ffi>=0.1.3,<0.1.12", # for supersonic_moe
]
Expand Down
2 changes: 1 addition & 1 deletion packages/paddlefleet_ops/third_party/cudnn-frontend
Submodule cudnn-frontend updated 970 files
Loading