Fixed fft test (use single device)
Reduce time spent in FFT test