🚀 The feature, motivation and pitch

Convolution operation is currently unsupported for complex types since cuDNN (the library we call into) doesn't support complex. However there have been several requests to add complex convolution. One of the recent requests was from an internal team who need complex convolution to multiply two polynomials (equivalent to convolution of their roots).

As a way to unblock people, we can start by implementing complex convolution as three real convolutions. And then, if needed, add specialized kernels for better performance.

• conv1d
• conv2d
• conv3d
• conv_transpose1d
• conv_transpose2d
• conv_transpose3d

Alternatives

No response

Additional context

No response

cc @ezyang @gchanan @zou3519 @bdhirsh @anjali411 @dylanbespalko @mruberry @lezcano @nikitaved