Skip to content

Code Contribution: 【Hard】【Operator Development】index_fill #892

@0x45f

Description

@0x45f

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.
请同时提供实现正确性测试与性能测试代码。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions