Code Contribution: 【Lv2】【Operator Development】argmin #318
Assignees
Comments
|
吴扬俊 认领 |
iclementine
pushed a commit
that referenced
this issue
Jan 10, 2025
* add argmin * support int dtype and int64 index * add test for dim=None * rebase master --------- Co-authored-by: wuyangjun <[email protected]>
Gxiandy
pushed a commit
to Gxiandy/FlagGems
that referenced
this issue
Jan 12, 2025
* add argmin * support int dtype and int64 index * add test for dim=None * rebase master --------- Co-authored-by: wuyangjun <[email protected]>
DuanYaQi
pushed a commit
that referenced
this issue
Jan 21, 2025
* add argmin * support int dtype and int64 index * add test for dim=None * rebase master --------- Co-authored-by: wuyangjun <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Description 任务介绍
Develop forward function for argmin operator.
开发argmin算子的前向功能。
Requirements 任务要求
Interface 接口
argmin(Tensor self, int? dim=None, bool keepdim=False) -> TensorFunction reference 功能参考
https://pytorch.org/docs/stable/generated/torch.argmin.html#torch-argmin
Implementation reference 实现参考
https://github.com/FlagOpen/FlagGems/blob/master/src/flag_gems/ops/argmax.py
The operator should support all optional arguments defined in the interface.
算子应支持接口中定义的所有参数选项。
DDL 提交时间
Please submit a Pull Request within 2 weeks after accepting the assignment.
请于接取任务后两周内提交PR。
Please provide both accuracy test and performance test code.
请同时提供实现正确性测试与性能测试代码。
The text was updated successfully, but these errors were encountered: