master

cann-robotfix: aclnnMultinomial 显式指定 TopK sortPolicy 对齐 GPU 索引顺序

beebc4f7Add on Sep 18History

Files	Last commit	Last update
aclnn_multinomial.cpp	fix: aclnnMultinomial 显式指定 TopK sortPolicy 对齐 GPU 索引顺序 Co-authored-by: huairuchen<chenhuairu@huawei.com> # message auto-generated for no-merge-commit merge: !5755 merge fix-multinomial-sortpolicy-align-gpu into master fix: aclnnMultinomial 显式指定 TopK sortPolicy 对齐 GPU 索引顺序 Created-by: huairuchen Commit-by: huairuchen Merged-by: cann-robot Description: ## 描述 本 PR 修复 aclnnMultinomial 算子（对应文件 random/stateless_sample_multinomial/op_api/aclnn_multinomial.cpp）在无放回采样场景下索引顺序与 GPU 不一致的问题。核心改动是在 Run950AicoreMultinomialWithoutReplacement 函数中，为 l0op::Topk 调用显式传入 sortPolicy 参数值 1，从而明确指定 TopK 的排序策略，使输出索引顺序对齐 GPU 侧行为。改动仅涉及一行参数调整，不影响其他逻辑分支。 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3227 ## 测试 冒烟测试通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5755	1 month 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