FFT operator in Relay IR

Questions
1

Does Relay have an FFT operator?

2

Found a way to add it to the operator registry, that works.

1 Like
3

Can you show me a simple code sample of how it was implemented?

4

https://tvm.apache.org/docs/dev/relay_add_op.html, you can modify this for your convenience.

1 Like
5

Thank you for reply!

6

Hi,

did you implement the FFT as a C /Python library and registered to tvm? or did you also define the FFT at the TE/TIR level and used TVM’s codegen to generate the implementation?