Flesh out most of AstGatherS

Mikolaj · Dec 26, 2024 · 56dab16 · 56dab16
1 parent 89416e4
commit 56dab16
Show file tree

Hide file tree

Showing 5 changed files with 364 additions and 35 deletions.
diff --git a/src/HordeAd/Core/Ast.hs b/src/HordeAd/Core/Ast.hs
@@ -459,7 +459,7 @@ data AstTensor :: AstMethodOfSharing -> AstSpanType -> TensorKindType
   AstFromVectorS :: (KnownNat n, KnownShS sh, TensorKind r)
                  => Data.Vector.Vector (AstTensor ms s (TKS2 sh r))
                  -> AstTensor ms s (TKS2 (n ': sh) r)
-  AstAppendS :: (KnownNat n, KnownNat m, KnownShS sh, TensorKind r)
+  AstAppendS :: (KnownNat m, KnownNat n, KnownShS sh, TensorKind r)
              => AstTensor ms s (TKS2 (m ': sh) r)
              -> AstTensor ms s (TKS2 (n ': sh) r)
              -> AstTensor ms s (TKS2 ((m + n) ': sh) r)