Remove the reshapes around the FFTs.
Remove reshapes around the CGemm and fix the tests.