From 4cddecca879736adee6c1fe004d7711e92b789c4 Mon Sep 17 00:00:00 2001 From: Mikolaj Konarski Date: Mon, 9 Feb 2026 17:11:51 +0100 Subject: Odds and ends --- src/Data/Array/Nested/Mixed.hs | 5 ++--- src/Data/Array/Nested/Shaped.hs | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) (limited to 'src/Data/Array/Nested') diff --git a/src/Data/Array/Nested/Mixed.hs b/src/Data/Array/Nested/Mixed.hs index 17f32b8..68d4498 100644 --- a/src/Data/Array/Nested/Mixed.hs +++ b/src/Data/Array/Nested/Mixed.hs @@ -57,7 +57,6 @@ import Data.Bag -- TODO: --- sumAllPrim :: (PrimElt a, NumElt a) => Mixed sh a -> a -- rminIndex1 :: Ranked (n + 1) a -> Ranked n Int -- gather/scatter-like things (most generally, the higher-order variants: accelerate's backpermute/permute) -- After benchmarking: matmul and matvec @@ -349,8 +348,8 @@ class Elt a where mtranspose :: forall is sh. (IsPermutation is, Rank is <= Rank sh) => Perm is -> Mixed sh a -> Mixed (PermutePrefix is sh) a - -- | All arrays in the input must have equal shapes, including subarrays - -- inside their elements. + -- | All arrays in the input must have equal shapes (except possibly + -- for the outermost dimension) including subarrays inside their elements. mconcat :: NonEmpty (Mixed (Nothing : sh) a) -> Mixed (Nothing : sh) a mrnf :: Mixed sh a -> () diff --git a/src/Data/Array/Nested/Shaped.hs b/src/Data/Array/Nested/Shaped.hs index b5937fc..08711b6 100644 --- a/src/Data/Array/Nested/Shaped.hs +++ b/src/Data/Array/Nested/Shaped.hs @@ -14,7 +14,7 @@ module Data.Array.Nested.Shaped ( liftShaped1, liftShaped2, ) where -import Prelude hiding (mappend, mconcat) +import Prelude hiding (mappend) import Data.Array.Internal.RankedG qualified as RG import Data.Array.Internal.RankedS qualified as RS -- cgit v1.2.3-70-g09d2