search

triton-lang / triton

Development repository for the Triton language and compiler
https://triton-lang.org/
MIT License
12.04k stars 1.43k forks source link

convolution sample #772

Open leiwen83 opened 1 year ago

leiwen83 commented 1 year ago

Hi,

Is there conv2d and transposed conv2d sample for triton?

Thx,

ptillet commented 1 year ago

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 commented 1 year ago

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 commented 1 year ago

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 commented 1 year ago

same question! :) @ptillet

huang-hz commented 8 months ago

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 commented 7 months ago

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