You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
请同时提供实现正确性测试与性能测试代码。
The text was updated successfully, but these errors were encountered:
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.
请同时提供实现正确性测试与性能测试代码。
The text was updated successfully, but these errors were encountered: