Good to know, I’ve edited the initial post. The timings are now unfortunately much worse:
FFT Conv Ele GPU Time: 4.759008884429932
FFT Conv Pruned GPU Time: 5.33543848991394
Functional Conv GPU Time: 0.07413554191589355
Using the same sizes for the CPU tests:
FFT Conv CPU Time: 66.0956494808197
Functional Conv CPU Time: 3.2627475261688232