Description 任务介绍
Develop forward function for index_fill operator.
开发index_fill算子的前向功能。
Requirements 任务要求
Interface 接口
index_fill.int_Scalar(Tensor self, int dim, Tensor index, Scalar value) -> Tensor
Function reference 功能参考
https://docs.pytorch.org/docs/stable/generated/torch.Tensor.index_fill.html
Implementation reference 实现参考
https://github.com/FlagOpen/FlagGems/blob/master/src/flag_gems/ops/index_add.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.
请同时提供实现正确性测试与性能测试代码。