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.
请同时提供实现正确性测试与性能测试代码。
Gxiandy
commented
1 week ago
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) -> TensorFunction 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.pyThe 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. 请同时提供实现正确性测试与性能测试代码。