master

文件	最后提交记录	最后更新时间
arch35	perf: float2/float4 for init_embedding_hash_table Co-authored-by: alfengyuan<yyfgdut@gmail.com> # message auto-generated for no-merge-commit merge: !9745 merge perf/init-bucket-b128 into master perf: float2/float4 for init_embedding_hash_table Created-by: alfengyuan Commit-by: alfengyuan Merged-by: cann-robot Description: ## 描述 InitEmbeddingHashTable 算子（arch35）引入 SIMT 访存合并优化：将桶头 24B 合并为 16B 常数写 + longlong2 写，values 区用 float4(B128) 循环批量写入代替逐元素标量访问，flag 字节折叠进 values 首块写窗。奇桶长 + dim≤4 回退标量路径。 **改动文件**： - op_kernel/arch35/init_embedding_hash_table.h：新增 InitBucket __simt_callee__ helper，重构 InitCompute 为宽写路径 + 标量回退路径（+85 -26） - op_host/arch35/init_embedding_hash_table_tiling_arch35.cpp：optional 输入改用 GetOptionalInputShape（+1 -1） **性能数据**（950PR，B=2^21，const 模式）： - dim≥25：~2× 带宽提升（~145 → ~288 GB/s） - dim=5~20：1.2~2.0× - dim=1~4：持平 - 全 dim 无回退 ## 关联的Issue 关联 Issue #5498 ## 测试 - e2e 真机精度 3/3 PASS（const dim=8 / const dim=48 / random dim=8 + lookup 交叉验证） - e2e 真机性能 35 组 dim（1~20, 25~100）全量验证，dim≥25 稳定 2× 带宽提升 - sweep 全场景矩阵 36/36 PASS ## 文档更新 无 ## 类型标签 - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!9745	1 个月前
CMakeLists.txt	feat: 为 init_embedding_hash_table 新增 ascend960dt 支持 Co-authored-by: alfengyuan<yyfgdut@gmail.com> # message auto-generated for no-merge-commit merge: !10742 merge feat/init-embedding-hash-table-ascend960dt into master feat: 为 init_embedding_hash_table 新增 ascend960dt 支持 Created-by: alfengyuan Commit-by: alfengyuan Merged-by: cann-robot Description: ## 描述 为 init_embedding_hash_table 算子新增 ascend960dt 芯片支持（参考现有 ascend950 配置）： - 新增 op_host/config/ascend960dt/init_embedding_hash_table_binary.json（内容与 ascend950 配置一致） - CMakeLists.txt：SUPPORT_COMPUTE_UNIT / SUPPORT_TILING_DIR 增加 ascend960dt（tiling 仍为 arch35） - op_host/init_embedding_hash_table_def.cpp：新增 AICore().AddConfig("ascend960dt", ...) - op_kernel/CMakeLists.txt：COMPUTE_UNITS 增加 ascend960dt 改动全部为增量添加，仅在 ASCEND_COMPUTE_UNIT=ascend960dt 构建时生效，不影响 ascend950 编译。 ## 关联的Issue 关联 #5987 ## 测试 本地未编译验证（环境无 CANN 依赖）。改动为纯增量配置，ascend950 构建路径不变。 ## 文档更新 无。 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [x] AI辅助编写 See merge request: cann/ops-nn!10742	18 天前
init_embedding_hash_table.cpp	revise kernel compile options Co-authored-by: cong-jiyu<congjiyu@h-partners.com> # message auto-generated for no-merge-commit merge: !5202 merge master into master revise kernel compile options Created-by: cong-jiyu Commit-by: cong-jiyu Merged-by: cann-robot Description: ## 描述 kernel配置项整改:foreach,hash,loss,rnn,vfusion ## 关联的Issue [#1985](https://gitcode.com/cann/ops-nn/issues/1985) ## 测试 本地编译，对比修改前后.o ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5202	4 个月前