| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
refactor clip_by_value/v2 with Brc_templates Co-authored-by: wangpengbo26<wangpengbo26@huawei.com> # message auto-generated for no-merge-commit merge: !3266 merge clipByValue into master refactor clip_by_value/v2 with Brc_templates Created-by: wangpengbo26 Commit-by: wangpengbo26 Merged-by: cann-robot Description: ## 描述 重构 conversion/clip_by_value 和 conversion/clip_by_value_v2 的 AICore 实现,以高性能 Broadcast 模板形式接入算子调度与计算流程。 主要改动: - Host 侧按 dtype 构造 ClipByValue/ClipByValueV2 的 OpDag,调用 BroadcastBaseTiling 完成模板 tiling,并根据 schMode 生成模板 tiling key。 - Kernel 侧新增 arch35 模板入口,通过 template <schMode> 和 BroadcastSch<schMode, OpDag> 执行广播调度。 - DAG 中融合 clip 的 min/max 两步计算,ClipByValue 保持 Min(x, max) 后 Max(result, min),ClipByValueV2 保持 Max(x, min) 后 Min(result, max)。 - int64 路径使用 MicroAPI RegTraitNumTwo 处理 64 位计算。 - 更新两个算子的 host tiling UT,覆盖 fp16、bf16、fp32、int32、int64、广播和 dtype mismatch 场景。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/2015 ## 测试 - ClipByValue 全量 CSV 精度用例 245 条最终 PASS。 - ClipByValueV2 全量 CSV 精度用例 169 条最终 PASS。 ## 文档更新 无。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:算子模板化重构 See merge request: cann/ops-math!3266 | 3 个月前 | |
refactor clip_by_value/v2 with Brc_templates Co-authored-by: wangpengbo26<wangpengbo26@huawei.com> # message auto-generated for no-merge-commit merge: !3266 merge clipByValue into master refactor clip_by_value/v2 with Brc_templates Created-by: wangpengbo26 Commit-by: wangpengbo26 Merged-by: cann-robot Description: ## 描述 重构 conversion/clip_by_value 和 conversion/clip_by_value_v2 的 AICore 实现,以高性能 Broadcast 模板形式接入算子调度与计算流程。 主要改动: - Host 侧按 dtype 构造 ClipByValue/ClipByValueV2 的 OpDag,调用 BroadcastBaseTiling 完成模板 tiling,并根据 schMode 生成模板 tiling key。 - Kernel 侧新增 arch35 模板入口,通过 template <schMode> 和 BroadcastSch<schMode, OpDag> 执行广播调度。 - DAG 中融合 clip 的 min/max 两步计算,ClipByValue 保持 Min(x, max) 后 Max(result, min),ClipByValueV2 保持 Max(x, min) 后 Min(result, max)。 - int64 路径使用 MicroAPI RegTraitNumTwo 处理 64 位计算。 - 更新两个算子的 host tiling UT,覆盖 fp16、bf16、fp32、int32、int64、广播和 dtype mismatch 场景。 ## 关联的Issue - https://gitcode.com/cann/ops-math/issues/2015 ## 测试 - ClipByValue 全量 CSV 精度用例 245 条最终 PASS。 - ClipByValueV2 全量 CSV 精度用例 169 条最终 PASS。 ## 文档更新 无。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [x] 性能优化 - [ ] 文档更新 - [x] 其他,请描述:算子模板化重构 See merge request: cann/ops-math!3266 | 3 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 3 个月前 | ||
| 3 个月前 |