Redesign GPU FFT abstraction