What should I do if the √N is not an integer?

[ChaiJinhao]

What should I do if the √N is not an integer?

[Edenzzzz]

Just find the closest integer factor. That's also in their code.

[ChaiJinhao]

Just find the closest integer factor. That's also in their code.

Thank you！！！

[DanFu09]

Yep, that's the right thing to do. We have a follow-up paper out now where we dive deeper into the tradeoffs for higher-order decompositions (focusing on the fixed FFT case, but many of the insights apply to the general case as well): https://github.com/HazyResearch/flash-fft-conv

We'll be moving the code into this codebase soon!