Fixed the issue with allocating fft_real and fft_imag. Previously, it was working for the smaller images but failing when processing the larger one. The error msg was 'unspecified launch failure'