search

tensor-compiler / taco

The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
http://tensor-compiler.org
Other
1.23k stars 184 forks source link

convolution #549

Open vnatesh opened 1 year ago

vnatesh commented 1 year ago

Is it possible to generate code for convolution operations? For example, for convolution with stride s, the tensor algebra expression O(m,h,w) = F(m,k,i,j) I(k, hs + i, w*s + j) is currently invalid in the online tool

rootjalex commented 1 year ago

A paper that built on TACO (and I believe appears in MLSys this year) has support for this type of convolution, the github repo is here: https://github.com/nullplay/Unified-Convolution-Framework

vnatesh commented 1 year ago

Thank you for the reference Alex!