Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Code Contribution: 【Lv3】【Operator Development】index_select_backward #316

Open
StrongSpoon opened this issue Nov 22, 2024 · 0 comments

Comments

@StrongSpoon
Copy link
Collaborator

StrongSpoon commented Nov 22, 2024

Description 任务介绍

Develop backward function for index_select operator.
开发index_select算子的反向功能。

Requirements 任务要求

Interface 接口
index_select_backward(Tensor grad, SymInt[] self_sizes, int dim, Tensor index) -> Tensor
Function reference 功能参考
https://pytorch.org/docs/stable/generated/torch.index_select.html#torch-index-select
Implementation reference 实现参考
https://github.com/FlagOpen/FlagGems/blob/master/src/flag_gems/ops/index_select.py

The operator should support all optional arguments defined in the interface.
算子应支持接口中定义的所有参数选项。

DDL 提交时间

Please submit a Pull Request within 3 weeks after accepting the assignment.
请于接取任务后三周内提交PR。
Please provide both accuracy test and performance test code.
请同时提供实现正确性测试与性能测试代码。

@StrongSpoon StrongSpoon converted this from a draft issue Nov 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: Todo
Development

No branches or pull requests

1 participant