From 0216dacb82f305e30f147ec7242dcd8599da721a Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Fri, 16 Jan 2026 19:14:59 +0100
Subject: Use numEltSumFull in X.sumFull

Thanks Mikolaj :)
---
 src/Data/Array/XArray.hs | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

(limited to 'src/Data/Array')

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
-- 
cgit v1.2.3-70-g09d2