choose the batch size of cuda fft