diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2026-01-16 19:14:59 +0100 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2026-01-16 19:15:12 +0100 |
| commit | 0216dacb82f305e30f147ec7242dcd8599da721a (patch) | |
| tree | 4ffde594b2c92a1ca778a2ed15e98ac3ab15f73e | |
| parent | 96795853db5a3ee85d7c838a508b4153988e6042 (diff) | |
Thanks Mikolaj :)
| -rw-r--r-- | src/Data/Array/XArray.hs | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/src/Data/Array/XArray.hs b/src/Data/Array/XArray.hs index 6389e67..1445ce6 100644 --- a/src/Data/Array/XArray.hs +++ b/src/Data/Array/XArray.hs @@ -268,11 +268,7 @@ transpose2 ssh1 ssh2 (XArray arr) = XArray (S.transpose (ssxIotaFrom ssh2 n1 ++ ssxIotaFrom ssh1 0) arr) sumFull :: (Storable a, NumElt a) => StaticShX sh -> XArray sh a -> a -sumFull _ (XArray arr) = - S.unScalar $ - liftO1 (numEltSum1Inner (SNat @0)) $ - S.fromVector [product (S.shapeL arr)] $ - S.toVector arr +sumFull ssx (XArray arr) = numEltSumFull (ssxRank ssx) $ fromO arr sumInner :: forall sh sh' a. (Storable a, NumElt a) => StaticShX sh -> StaticShX sh' -> XArray (sh ++ sh') a -> XArray sh a |