Star0
0
Code
Code
Issues
Pull Requests
Actions
Discussion
Wiki
Analysis
Star0
0
cann-robotcann-robot[ascend960] bernoulli/choicewithmask/drop_out_do_mask_v3(_d)/stateless_randperm 算子新增 ascend960dt/ascend960pr 平台支持
8ade3796Add on 24 days agoHistory
FilesLast commitLast update
aclnn_randperm.cpp
[ascend960] bernoulli/choicewithmask/drop_out_do_mask_v3(_d)/stateless_randperm 算子新增 ascend960dt/ascend960pr 平台支持 Co-authored-by: liangtongxue<lianglihui@h-partners.com> # message auto-generated for no-merge-commit merge: !6151 merge stateless-bernoulli-ascend960 into master [ascend960] bernoulli/choicewithmask/drop_out_do_mask_v3(_d)/stateless_randperm 算子新增 ascend960dt/ascend960pr 平台支持 Created-by: liangtongxue Commit-by: liangtongxue Merged-by: cann-robot Description: ## 描述 stateless_bernoulli、stateless_random_choice_with_mask、drop_out_do_mask_v3、drop_out_do_mask_v3_d、stateless_randperm 五个 random 算子新增 ascend960dt / ascend960pr 平台支持，补齐 960 公共构建配置（对齐 !6074 清单）与 bernoulli 的 L2 aclnn 层适配。v3 系列适配点合入自 !6186（zyf0712）、randperm 合入自 !6144（guankarl），均经 review 并补齐遗漏项。**本次适配不涉及资料与 ST 用例，后续单独处理。** L2/L0 芯片路由统一使用底层 IsRegBase()（common/inc/op_api/aclnn_check.h 已含 DAV_3510/9201/9202）。 ### stateless_bernoulli（A6 适配点合入 + L2 层补齐） - 以 A5 主线 random_common 框架版为主体（A6 唯一 kernel 适配点 PHILOX_THREAD_LAUNCH=512 已含于主线重构版），tiling 沿用 arch35 - **L2 层**：统一使用底层 IsRegBase() 覆盖四处分支（dtype 列表×2 / 张量 prob Cast float / 标量 prob 转 float tensor），960 与 950 走同一 L0 路径且行为一致 ### stateless_random_choice_with_mask（基于 A5 现有实现适配，A6 暂无源码） - kernel 核数/线程/UB 全部经 tilingData 动态下发（无写死核数）；Reg:: / asc_vf_call / PhiloxRandom<10> 均有 960 先例；tiling 平台参数全部动态获取；全链路无 SocVersion 区间判断 ### drop_out_do_mask_v3 / drop_out_do_mask_v3_d（合入 !6186，已 review） - tiling 新增 BeforeProcess 按 NpuArch 动态设置 coreAlignSize：960 用 32（DV121），其余维持 256。**已核验 BeforeProcess 早于 coreAlignSize 全部消费点，950/350/910b 行为与原静态设置等价** - kernel 零改动；binary.json 与 ascend950 一致；v3_d 依赖已闭环 ### stateless_randperm（合入 !6144 + 补齐其遗漏） - def / CMakeLists / config(ascend960dt+pr) / ascendc_config 登记（auto_sync=true 存量风格） - **L0/L2 路由补 960**（!6144 遗漏）：CheckAiCoreIsSupport 原仅 DAV_3510 会误走 AICPU，统一改用 IsRegBase()；L2 bf16 同补 - **kernel const 兼容**：GET_TILING_DATA_WITH_STRUCT 产生 const tilingData，构造/成员/reinterpret_cast 全链 const 化（对齐 split_v 先例，950 双向兼容） - 依赖说明：kernel 内嵌 sort 计算（非运行时依赖 sort 算子） ### 公共构建配置（与基线取并集） build.conf.sh / build_options.sh / build_env.sh 净增 960pr（960dt 基线已含）；random_common 960 已由 !6074 合入基线；ascendc_config.json 登记五算子 960 ### 待定项（后续单独处理） - fusion pass（StatelessBernoulliV2 融合）CheckPlatform 白名单 - README/docs 资料与 ST 用例 ### 950 影响评估 全部为追加式/条件扩展式改动：kernel/tiling 950 既有路径零改动、L2/L0 条件扩展不改变 950 判断结果。**950 出包行为不变。** ## 关联的Issue #3623 ## 测试 - pre-commit 门禁全部通过 - bernoulli / drop_out_do_mask_v3 / v3_d 已完成 960 单算子验证；choicewithmask 已出包验证 - 950 出包行为不变 ## 文档更新 不涉及（资料后续单独处理） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6151
24 days ago
aclnn_randperm.h
950支持stateless_randperm Co-authored-by: gaojie120<gaojie120@huawei.com> Co-authored-by: CXMT<liubaoyang3@huawei.com> Co-authored-by: littlemons<yaowenjie7@huawei.com> Co-authored-by: ethWeek1<penghaohang@huawei.com> # message auto-generated for no-merge-commit merge: !1211 merge master into master 950支持stateless_randperm Created-by: ethWeek1 Commit-by: ethWeek1;littlemons;CXMT;gaojie120 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 新增stateless_randperm ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#776](https://gitcode.com/cann/ops-math/issues/776) <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!1211
8 months ago
stateless_randperm.cpp
[ascend960] bernoulli/choicewithmask/drop_out_do_mask_v3(_d)/stateless_randperm 算子新增 ascend960dt/ascend960pr 平台支持 Co-authored-by: liangtongxue<lianglihui@h-partners.com> # message auto-generated for no-merge-commit merge: !6151 merge stateless-bernoulli-ascend960 into master [ascend960] bernoulli/choicewithmask/drop_out_do_mask_v3(_d)/stateless_randperm 算子新增 ascend960dt/ascend960pr 平台支持 Created-by: liangtongxue Commit-by: liangtongxue Merged-by: cann-robot Description: ## 描述 stateless_bernoulli、stateless_random_choice_with_mask、drop_out_do_mask_v3、drop_out_do_mask_v3_d、stateless_randperm 五个 random 算子新增 ascend960dt / ascend960pr 平台支持，补齐 960 公共构建配置（对齐 !6074 清单）与 bernoulli 的 L2 aclnn 层适配。v3 系列适配点合入自 !6186（zyf0712）、randperm 合入自 !6144（guankarl），均经 review 并补齐遗漏项。**本次适配不涉及资料与 ST 用例，后续单独处理。** L2/L0 芯片路由统一使用底层 IsRegBase()（common/inc/op_api/aclnn_check.h 已含 DAV_3510/9201/9202）。 ### stateless_bernoulli（A6 适配点合入 + L2 层补齐） - 以 A5 主线 random_common 框架版为主体（A6 唯一 kernel 适配点 PHILOX_THREAD_LAUNCH=512 已含于主线重构版），tiling 沿用 arch35 - **L2 层**：统一使用底层 IsRegBase() 覆盖四处分支（dtype 列表×2 / 张量 prob Cast float / 标量 prob 转 float tensor），960 与 950 走同一 L0 路径且行为一致 ### stateless_random_choice_with_mask（基于 A5 现有实现适配，A6 暂无源码） - kernel 核数/线程/UB 全部经 tilingData 动态下发（无写死核数）；Reg:: / asc_vf_call / PhiloxRandom<10> 均有 960 先例；tiling 平台参数全部动态获取；全链路无 SocVersion 区间判断 ### drop_out_do_mask_v3 / drop_out_do_mask_v3_d（合入 !6186，已 review） - tiling 新增 BeforeProcess 按 NpuArch 动态设置 coreAlignSize：960 用 32（DV121），其余维持 256。**已核验 BeforeProcess 早于 coreAlignSize 全部消费点，950/350/910b 行为与原静态设置等价** - kernel 零改动；binary.json 与 ascend950 一致；v3_d 依赖已闭环 ### stateless_randperm（合入 !6144 + 补齐其遗漏） - def / CMakeLists / config(ascend960dt+pr) / ascendc_config 登记（auto_sync=true 存量风格） - **L0/L2 路由补 960**（!6144 遗漏）：CheckAiCoreIsSupport 原仅 DAV_3510 会误走 AICPU，统一改用 IsRegBase()；L2 bf16 同补 - **kernel const 兼容**：GET_TILING_DATA_WITH_STRUCT 产生 const tilingData，构造/成员/reinterpret_cast 全链 const 化（对齐 split_v 先例，950 双向兼容） - 依赖说明：kernel 内嵌 sort 计算（非运行时依赖 sort 算子） ### 公共构建配置（与基线取并集） build.conf.sh / build_options.sh / build_env.sh 净增 960pr（960dt 基线已含）；random_common 960 已由 !6074 合入基线；ascendc_config.json 登记五算子 960 ### 待定项（后续单独处理） - fusion pass（StatelessBernoulliV2 融合）CheckPlatform 白名单 - README/docs 资料与 ST 用例 ### 950 影响评估 全部为追加式/条件扩展式改动：kernel/tiling 950 既有路径零改动、L2/L0 条件扩展不改变 950 判断结果。**950 出包行为不变。** ## 关联的Issue #3623 ## 测试 - pre-commit 门禁全部通过 - bernoulli / drop_out_do_mask_v3 / v3_d 已完成 960 单算子验证；choicewithmask 已出包验证 - 950 出包行为不变 ## 文档更新 不涉及（资料后续单独处理） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6151
24 days ago
stateless_randperm.h
950支持stateless_randperm Co-authored-by: gaojie120<gaojie120@huawei.com> Co-authored-by: CXMT<liubaoyang3@huawei.com> Co-authored-by: littlemons<yaowenjie7@huawei.com> Co-authored-by: ethWeek1<penghaohang@huawei.com> # message auto-generated for no-merge-commit merge: !1211 merge master into master 950支持stateless_randperm Created-by: ethWeek1 Commit-by: ethWeek1;littlemons;CXMT;gaojie120 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 新增stateless_randperm ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> [#776](https://gitcode.com/cann/ops-math/issues/776) <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!1211
8 months ago