Support for staged distributed FFTs