已合并
add argmax(min)withvalue attr for dtype #1049
add argmax(min)withvalue attr for dtype #1049
已合并
lu-yukun创建于 1月31日
lu-yukun
1月31日

描述

add argmax(min)withvalue attr for dtype

关联的Issue

https://gitcode.com/cann/ops-math/issues/642

测试

文档更新

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 lu-yukun 的贡献)
CANN-robot
CANN-robot成员
1月31日 评论:

Thanks for your pull-request.

The full list of commands accepted by me can be found at here.

You can get sig-info at here

likedislike
CANN-robot
CANN-robot成员
1月31日 评论:

以下是根据您提交的修改文件推荐的Reviewer和Committer序列,需各模块评审通过后方可合入

Module List Reviewers Committers
*/*/op_graph/*_proto.h N/A wangyongguang, tang-lei01
likedislike
CANN-robotCANN-robot成员
1月31日 添加了label:cann-cla/yes
CANN-robot
CANN-robot成员
1月31日 评论:

🔵 source code change are detected, tasks labels is removed in this pull request!

likedislike
CANN-robot
CANN-robot成员
1月31日 评论:

问题/功能描述

本次PR为ArgMaxWithValue和ArgMinWithValue算子新增了可选的indice_dtype属性,以解决算子输出索引数据类型固定为DT_INT32的问题。该增强使得算子能够灵活输出DT_INT64类型的索引,从而更好地兼容需要更大索引范围的场景(如PyTorch框架对接),提升了算子的灵活性和跨框架兼容性。

修改方案描述

在ArgMaxWithValue和ArgMinWithValue两个算子的原型定义中,分别增加了一个名为indice_dtype、类型为Type的属性,并将其默认值设置为DT_INT32。该方案允许用户在调用算子时显式指定输出索引的数据类型(DT_INT32DT_INT64),而无需修改算子内核或依赖外部转换,简化了使用流程并增强了接口的易用性。

likedislike
此处折叠了66条消息 查看更多
CANN-robot
CANN-robot成员
2月4日 评论:

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
Coopermassaki

likedislike
CANN-robotCANN-robot成员
2月4日 添加了label:lgtmapproved
CANN-robot
CANN-robot成员
2月4日 评论:

Review Guide

This Pull-Request Passes Review.
Committers who wrote a comment of /approve are: wangyongguang, zhou-qilong.
Reviewers who wrote a comment of /lgtm are: wangrui_, zhou-qilong, wangyongguang.

likedislike
CANN-robotCANN-robot成员
2月4日 关闭了关联的issue
CANN-robotCANN-robot成员
2月4日 合入了pull request