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: 【Lv2】【Operator Development】cummin #321

Open
StrongSpoon opened this issue Nov 22, 2024 · 1 comment · May be fixed by #339
Open

Code Contribution: 【Lv2】【Operator Development】cummin #321

StrongSpoon opened this issue Nov 22, 2024 · 1 comment · May be fixed by #339
Assignees

Comments

@StrongSpoon
Copy link
Collaborator

StrongSpoon commented Nov 22, 2024

Description 任务介绍

Develop forward function for cummin operator.
开发cummin算子的前向功能。

Requirements 任务要求

Interface 接口
cummin(Tensor self, int dim) -> (Tensor values, Tensor indices)
Function reference 功能参考
https://pytorch.org/docs/stable/generated/torch.cummin.html#torch-cummin
Implementation reference 实现参考
https://github.com/FlagOpen/FlagGems/blob/master/src/flag_gems/ops/cumsum.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.
请同时提供实现正确性测试与性能测试代码。

@StrongSpoon StrongSpoon converted this from a draft issue Nov 22, 2024
@awayzjj
Copy link
Collaborator

awayzjj commented Nov 22, 2024

Binary 认领

@awayzjj awayzjj linked a pull request Dec 2, 2024 that will close this issue
3 tasks
@Tango2018cc Tango2018cc moved this from In Progress to Review in Triton China Community Dec 4, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: Review
Development

Successfully merging a pull request may close this issue.

2 participants