Closed
[feat][ops]fa_support_32_64_128_256_dim #244
cwh19980101Add on 4 days agoClosed at 2 days ago
[feat][ops]fa_support_32_64_128_256_dim #244
Closed
Merged fromcwh1998/ops-recfa_multi_dimsintoAscend/ops-recmaster
The current Pull Request has been closed by cwh19980101.
4 days ago Associated milestone: ops-rec 26.3.0
4 days ago Added label: pr-audit-failed
xiangjie10
4 days ago Comments：
4 days ago Comments：
🔍 PR 规范审计未通过，以下项目需要修正：
- ❌ PR 新增代码 1946 行超过 1000 行，且标题未标注"反合"
请修正后重新提交，或联系仓库管理员。
4 days ago Modify description of pull request
4 days ago Added label: ascend-cla/yes
153 messages are collapsed here。 See more
3 days ago Added label: docs-ci-pipeline-success
3 days ago Removed label: ci-pipeline-running
3 days ago Added label: ci-pipeline-passed
AtlasAccount
3 days ago Comments：
3 days ago Comments：
流水线 PR-pipeline_ops-rec#1153 [ commitID：be5ef75b ] 已完成
>>>代码风格自动修复执行成功（无修复内容）
| 阶段 | 任务名 | 状态 | 详情 |
|---|---|---|---|
| 编译构建 | PR-build-SAST-check | ✅ COMPLETED | >>> |
| Build-2.7.1_arm | ✅ COMPLETED | >>> | |
| Build-2.10.0_arm | ✅ COMPLETED | >>> | |
| Build-2.7.1_X86 | ✅ COMPLETED | >>> | |
| Build-2.10.0_X86 | ✅ COMPLETED | >>> | |
| 恶意代码检查 | Antipoison | ✅ COMPLETED | >>> |
| 编码安全与规范检查 | pre-commit | ✅ COMPLETED | >>> |
| clang-tidy | ✅ COMPLETED | >>> | |
| 开源片段检查 | SCA | ✅ COMPLETED | >>> |
| 开发者测试 | presmoke | ✅ COMPLETED | >>> |
| 流水线 | PR-pipeline_ops-rec | ✅ COMPLETED | >>> |
- compile : 运行流水线
- retry : 重试流水线所有失败子任务
- retry <任务名> : 仅重试指定失败子任务
- stop : 停止流水线
2 days ago Close pull request
合入来源
问题/功能描述
任意mask的FA算子支持32/64/128/256 embedding
子PR：https://gitcode.com/Ascend/ops-rec/pull/246
子PR：https://gitcode.com/Ascend/ops-rec/pull/247
修改方案描述
本地化catlass组件，支持动态embed32/64/128/256 embed,分为128/256档
128档：处理32/64/128 embed
256档：处理256 embed， n-chunk，256档由于UB不够，使用2 chunk，每次处理128 embed，共同组成256 embed结果。
是否涉及UT/ST
开发自检