master

文件	最后提交记录	最后更新时间
test_aclnn_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	1 天前
test_aclnn_multinomial_tensor.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	1 个月前