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】 conv3d #312

Open
StrongSpoon opened this issue Nov 22, 2024 · 1 comment
Open

Code Contribution: 【Lv3】【Operator Development】 conv3d #312

StrongSpoon opened this issue Nov 22, 2024 · 1 comment
Assignees

Comments

@StrongSpoon
Copy link
Collaborator

StrongSpoon commented Nov 22, 2024

Description 任务介绍

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

Requirements 任务要求

Interface 接口
conv3d(Tensor input, Tensor weight, Tensor? bias=None, SymInt[3] stride=1, SymInt[3] padding=0, SymInt[3] dilation=1, SymInt groups=1) -> Tensor
Function reference 功能参考
https://pytorch.org/docs/stable/generated/torch.nn.Conv3d.html
Implementation reference 实现参考
https://github.com/FlagOpen/FlagGems/blob/7016596a5071a3f2a969e19034e7dae58b699b36/src/flag_gems/ops/conv2d.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
@Gxiandy
Copy link
Contributor

Gxiandy commented Nov 22, 2024

Gxian认领

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: In Progress
Development

No branches or pull requests

2 participants