convolution sample

[leiwen83]

Hi,

Is there conv2d and transposed conv2d sample for triton?

Thx,

[ptillet]

There is no conv2d sample right now, but torch-inductor may have one? (the triton compiler doesn't do a very good job on it though -- we'll look into that in a few months)

[Jokeren]

https://github.com/pytorch/torchdynamo/blob/main/torchinductor/triton_ops/conv.py https://github.com/pytorch/torchdynamo/blob/main/torchinductor/triton_ops/conv1x1.py

[georgemorgan]

Has there been any progress on identifying why the compiler doesn't do a great job on the convolutional kernels yet? Also curious if there are any new reference samples on convolutional kernels in Triton.

[RuABraun]

same question! :) @ptillet

[huang-hz]

the support of conv have some questions , conv ops has been deleted in pytorch repo right now. The history version can't run rightly.

[krishvishal]

@huang-hz It has been moved to here: https://github.com/pytorch/pytorch/blob/main/torch/_inductor/kernel/conv.py