master

cann-robot[feat] ChunkCat, StatelessSampleMultinonmial适配ascend960dt

5f17d2e2Add on 19 days agoHistory

Files	Last commit	Last update
aclnn_multinomial.cpp	[feat] ChunkCat, StatelessSampleMultinonmial适配ascend960dt Co-authored-by: wsunmoon<wangming176@huawei.com> # message auto-generated for no-merge-commit merge: !6430 merge chunkcatMultinomial into master [feat] ChunkCat, StatelessSampleMultinonmial适配ascend960dt Created-by: wsunmoon Commit-by: wsunmoon Merged-by: cann-robot Description: ## 描述 ChunkCat, StatelessSampleMultinonmial算子适配960DT: 新增960DT二进制配置json文件 CMakeLists.txt文件配置960DT 算子信息库新增960DT aclnn接口新增支持960DT ChunkCat算子在960DT上做先分多核再分UB的性能优化 ## 关联的Issue 关联Issue [#3763](https://gitcode.com/cann/ops-math/issues/3763) ## 测试 UT, ST, 冒烟通过 ## 文档更新 不涉及 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 - [ ] AI辅助编写 See merge request: cann/ops-math!6430	19 days ago
aclnn_multinomial.h	fix: align multinomial sampling with normalized probabilities Co-authored-by: fenglin28<fenglin28@huawei.com> # message auto-generated for no-merge-commit merge: !4500 merge fix/stateless-sample-multinomial-assert into master fix: align multinomial sampling with normalized probabilities Created-by: fenglin28 Commit-by: fenglin28 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复 StatelessSampleMultinomial 边界场景下的采样行为： - 在二分查找前检查 CDF 最后一项是否大于 0 - 新增可选输入 norm_probs，用于准确判断零概率类别 - norm_probs 存在时，使用 x 执行二分查找，使用 norm_probs 处理零概率类别回退；缺省时保留基于相邻 CDF 值的兼容路径 - 增加 x 的维度校验，以及 x 与 norm_probs 的形状一致性校验 - 同步更新算子定义、L0 接口、tiling、kernel 配置、UT 和 README ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2558 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 完整通过 UT、泛化 ST 及流水线验证 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 更新 README 中 x 和 norm_probs 的输入及约束说明 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4500	2 months ago
multinomial_with_replacement.cpp	feat: adapt multinomial with stateless_sample_multinomial Co-authored-by: fenglin28<fenglin28@huawei.com> # message auto-generated for no-merge-commit merge: !3127 merge multinomial-torch-compatible into master feat: adapt multinomial with stateless_sample_multinomial Created-by: fenglin28 Commit-by: fenglin28 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 适配 multinomial 算子并新增 stateless_sample_multinomial ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1991 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 完整通过测试流水线 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 更新op_api_list.md 迁移aclnnMultinomial.md 迁移aclnnMultinomialTensor.md 新增stateless_sample_multinomial目录下README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3127	4 months ago
multinomial_with_replacement.h	feat: adapt multinomial with stateless_sample_multinomial Co-authored-by: fenglin28<fenglin28@huawei.com> # message auto-generated for no-merge-commit merge: !3127 merge multinomial-torch-compatible into master feat: adapt multinomial with stateless_sample_multinomial Created-by: fenglin28 Commit-by: fenglin28 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 适配 multinomial 算子并新增 stateless_sample_multinomial ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1991 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 完整通过测试流水线 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 更新op_api_list.md 迁移aclnnMultinomial.md 迁移aclnnMultinomialTensor.md 新增stateless_sample_multinomial目录下README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!3127	4 months ago
stateless_sample_multinomial.cpp	fix: align multinomial sampling with normalized probabilities Co-authored-by: fenglin28<fenglin28@huawei.com> # message auto-generated for no-merge-commit merge: !4500 merge fix/stateless-sample-multinomial-assert into master fix: align multinomial sampling with normalized probabilities Created-by: fenglin28 Commit-by: fenglin28 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复 StatelessSampleMultinomial 边界场景下的采样行为： - 在二分查找前检查 CDF 最后一项是否大于 0 - 新增可选输入 norm_probs，用于准确判断零概率类别 - norm_probs 存在时，使用 x 执行二分查找，使用 norm_probs 处理零概率类别回退；缺省时保留基于相邻 CDF 值的兼容路径 - 增加 x 的维度校验，以及 x 与 norm_probs 的形状一致性校验 - 同步更新算子定义、L0 接口、tiling、kernel 配置、UT 和 README ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2558 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 完整通过 UT、泛化 ST 及流水线验证 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 更新 README 中 x 和 norm_probs 的输入及约束说明 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4500	2 months ago
stateless_sample_multinomial.h	fix: align multinomial sampling with normalized probabilities Co-authored-by: fenglin28<fenglin28@huawei.com> # message auto-generated for no-merge-commit merge: !4500 merge fix/stateless-sample-multinomial-assert into master fix: align multinomial sampling with normalized probabilities Created-by: fenglin28 Commit-by: fenglin28 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复 StatelessSampleMultinomial 边界场景下的采样行为： - 在二分查找前检查 CDF 最后一项是否大于 0 - 新增可选输入 norm_probs，用于准确判断零概率类别 - norm_probs 存在时，使用 x 执行二分查找，使用 norm_probs 处理零概率类别回退；缺省时保留基于相邻 CDF 值的兼容路径 - 增加 x 的维度校验，以及 x 与 norm_probs 的形状一致性校验 - 同步更新算子定义、L0 接口、tiling、kernel 配置、UT 和 README ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/2558 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 完整通过 UT、泛化 ST 及流水线验证 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 更新 README 中 x 和 norm_probs 的输入及约束说明 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!4500	2 months ago