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】gather_backward #317

Closed
StrongSpoon opened this issue Nov 22, 2024 · 2 comments
Closed
Assignees

Comments

@StrongSpoon
Copy link
Collaborator

StrongSpoon commented Nov 22, 2024

Description 任务介绍

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

Requirements 任务要求

Interface 接口
gather_backward(Tensor grad, Tensor self, int dim, Tensor index, bool sparse_grad) -> Tensor
Function reference 功能参考
https://pytorch.org/docs/stable/generated/torch.gather.html#torch-gather
Implementation reference 实现参考
https://github.com/FlagOpen/FlagGems/blob/master/src/flag_gems/ops/gather.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
@awayzjj
Copy link
Collaborator

awayzjj commented Dec 5, 2024

Binary 认领

@Tango2018cc Tango2018cc moved this from Todo to In Progress in Triton China Community Dec 6, 2024
@awayzjj awayzjj mentioned this issue Dec 15, 2024
3 tasks
@Tango2018cc Tango2018cc moved this from In Progress to Review in Triton China Community Dec 16, 2024
@awayzjj
Copy link
Collaborator

awayzjj commented Jan 3, 2025

This issue could be closed? since it was completed in #382 @StrongSpoon @Tango2018cc

@Tango2018cc Tango2018cc moved this from Review to Done in Triton China Community Jan 8, 2025
@Tango2018cc Tango2018cc closed this as completed by moving to Done in Triton China Community Jan 8, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: Done
Development

Successfully merging a pull request may close this issue.

3 participants