NVIDIA GPU上的最优2D FFT尺寸探究

我们在NVIDIA A100上开展2D FFT性能基准测试，以确定哪些尺寸具备最佳性能。测试执行方式如下：每次测试不同尺寸后都会清理GPU内存。

def run_fft():
    fft2(array, axes=(-2, -1), overwrite_x=True)
timing = cupyx.timing.repeat(run_fft, repeat=10, n_warmup=1)

对一系列可能的尺寸进行测试后得到如下测量结果：

可以看到，存在一部分尺寸的运行速度慢于其他尺寸（主序列上方的准线性条纹），这些尺寸包含以低质数（如2和3）为因子的数值。请问是否存在通用规则来定义能实现最优运行的2D FFT尺寸（例如在CPU场景下使用fftw3时，有对应的通用规则定义）？

内容的提问来源于stack exchange，提问作者lessju