Enabled n-dimensional redistribution/local FFT in cuda code