Another place dimensions should be reversed

AccelerateHS · Aug 27, 2014 · 7924c8a · 7924c8a
1 parent bf73d55
commit 7924c8a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Data/Array/Accelerate/Math/FFT.hs b/Data/Array/Accelerate/Math/FFT.hs
@@ -274,8 +274,8 @@ cudaFFT mode sh = cudaFFT'
     hndl = unsafePerformIO $ do
             plan <- case shapeToList sh of
                      [width]                -> plan1D              width types 1
-                     [height, width]        -> plan2D       height width types
-                     [depth, height, width] -> plan3D depth height width types
+                     [width, height]        -> plan2D       height width types
+                     [width, height, depth] -> plan3D depth height width types
                      _                      -> error "Accelerate-fft cannot use CUFFT for arrays of dimensions higher than 3"
             addFinalizer plan (destroy plan)
             return plan