fix!

Mikolaj · Nov 16, 2024 · 4d8a350 · 4d8a350
1 parent 70ff615
commit 4d8a350
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/src/HordeAd/Core/AstSimplify.hs b/src/HordeAd/Core/AstSimplify.hs
@@ -1928,7 +1928,7 @@ astTransposeS perm t = case perm of
                  :: Permutation.PermutePrefix perm3 sh2
                     :~: Permutation.PermutePrefix perm sh) $
       case compare (length perm3V)
-                   (Nested.Internal.Shape.shsSize (knownShS @sh2)) of
+                   (Nested.Internal.Shape.shsLength (knownShS @sh2)) of
         LT -> gcastWith (unsafeCoerce Refl
                          :: Compare (Rank perm3) (Rank sh2) :~: LT) $
               astTransposeS perm3 u

diff --git a/src/HordeAd/Core/AstTools.hs b/src/HordeAd/Core/AstTools.hs
@@ -122,8 +122,8 @@ shapeAstFull t = case t of
   AstReverseS{} -> FTKS knownShS
   AstTransposeS @perm @sh2 perm _v ->
     withShapeP
-      (permutePrefixList (Permutation.permToList' perm)
-                         (shapeT @sh2)) $ \(Proxy @sh2Perm) ->
+      (backpermutePrefixList (Permutation.permToList' perm)
+                             (shapeT @sh2)) $ \(Proxy @sh2Perm) ->
         gcastWith (unsafeCoerce Refl :: sh2Perm :~: Permutation.PermutePrefix perm sh2) $
         FTKS knownShS
   AstReshapeS{} -> FTKS knownShS

diff --git a/src/HordeAd/Core/Delta.hs b/src/HordeAd/Core/Delta.hs
@@ -676,8 +676,8 @@ shapeDeltaFull = \case
   ReverseS{} -> FTKS knownShS
   TransposeS @perm @sh2 perm _v ->
     withShapeP
-      (permutePrefixList (Permutation.permToList' perm)
-                         (shapeT @sh2)) $ \(Proxy @sh2Perm) ->
+      (backpermutePrefixList (Permutation.permToList' perm)
+                             (shapeT @sh2)) $ \(Proxy @sh2Perm) ->
         gcastWith (unsafeCoerce Refl :: sh2Perm :~: Permutation.PermutePrefix perm sh2) $
         FTKS knownShS
   ReshapeS{} -> FTKS knownShS