About convolution

[Shaquille-Wu]

Please describe your question

Hello, FlagGems experts:
I didn't find the ops about convolution in FlagGems
Is there anyone would like to tell me where the "convolution" is?

[AKatydid]

Hello, I've been using conv ops lately, is this link below what you need?

https://github.com/FlagOpen/FlagGems/blob/master/src/flag_gems/ops/conv2d.py

[StrongSpoon]

triton convolution is not registered into aten library yet, since its performance is still worse than aten. if you want to use it, call it by flag_gems.conv1d, flag_gems.conv2d or flag_gems._conv_depthwise2d.

[qtlark]

Hello, I've been using conv ops lately, is this link below what you need?

https://github.com/FlagOpen/FlagGems/blob/master/src/flag_gems/ops/conv2d.py

It seems like there is still no grad for wight and bias

[StrongSpoon]

Hello, I've been using conv ops lately, is this link below what you need?
https://github.com/FlagOpen/FlagGems/blob/master/src/flag_gems/ops/conv2d.py

It seems like there is still no grad for wight and bias

convolutino_backward is in pr #365 and it's unfinished yet.