From 1f44ef9ce2e6cee4f71b044ffe0b01cd37603790 Mon Sep 17 00:00:00 2001
From: Mikolaj Konarski <mikolaj.konarski@funktory.com>
Date: Thu, 25 Dec 2025 23:45:41 +0100
Subject: Add pragmas to functions that -fprof-late-overloaded-calls reports

as breaking the chain of specialization (but it's currently broken,
so these may be false positives, but these pragmas may be handy
in the future anyway and they can't do much harm in theory
and in practice they are right now neutral performance-wise).
---
 src/Data/Array/Nested/Shaped.hs | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'src/Data/Array/Nested/Shaped.hs')

diff --git a/src/Data/Array/Nested/Shaped.hs b/src/Data/Array/Nested/Shaped.hs
index 5c52220..36ef24a 100644
--- a/src/Data/Array/Nested/Shaped.hs
+++ b/src/Data/Array/Nested/Shaped.hs
@@ -96,17 +96,21 @@ slift2 :: forall sh1 sh2 sh3 a. Elt a
        -> Shaped sh1 a -> Shaped sh2 a -> Shaped sh3 a
 slift2 sh3 f (Shaped arr1) (Shaped arr2) = Shaped (mlift2 (ssxFromShX (shxFromShS sh3)) f arr1 arr2)
 
+{-# INLINE ssumOuter1PrimP #-}
 ssumOuter1PrimP :: forall sh n a. (Storable a, NumElt a)
                 => Shaped (n : sh) (Primitive a) -> Shaped sh (Primitive a)
 ssumOuter1PrimP (Shaped arr) = Shaped (msumOuter1PrimP arr)
 
+{-# INLINEABLE ssumOuter1Prim #-}
 ssumOuter1Prim :: forall sh n a. (NumElt a, PrimElt a)
                => Shaped (n : sh) a -> Shaped sh a
 ssumOuter1Prim = sfromPrimitive . ssumOuter1PrimP . stoPrimitive
 
+{-# INLINE ssumAllPrimP #-}
 ssumAllPrimP :: (PrimElt a, NumElt a) => Shaped n (Primitive a) -> a
 ssumAllPrimP (Shaped arr) = msumAllPrimP arr
 
+{-# INLINE ssumAllPrim #-}
 ssumAllPrim :: (PrimElt a, NumElt a) => Shaped n a -> a
 ssumAllPrim (Shaped arr) = msumAllPrim arr
 
@@ -126,15 +130,19 @@ sappend = coerce mappend
 sscalar :: Elt a => a -> Shaped '[] a
 sscalar x = Shaped (mscalar x)
 
+{-# INLINEABLE sfromVectorP #-}
 sfromVectorP :: Storable a => ShS sh -> VS.Vector a -> Shaped sh (Primitive a)
 sfromVectorP sh v = Shaped (mfromVectorP (shxFromShS sh) v)
 
+{-# INLINEABLE sfromVector #-}
 sfromVector :: PrimElt a => ShS sh -> VS.Vector a -> Shaped sh a
 sfromVector sh v = sfromPrimitive (sfromVectorP sh v)
 
+{-# INLINEABLE stoVectorP #-}
 stoVectorP :: Storable a => Shaped sh (Primitive a) -> VS.Vector a
 stoVectorP = coerce mtoVectorP
 
+{-# INLINEABLE stoVector #-}
 stoVector :: PrimElt a => Shaped sh a -> VS.Vector a
 stoVector = coerce mtoVector
 
@@ -261,6 +269,7 @@ sminIndexPrim (Shaped arr) = ixsFromIxX (mminIndexPrim arr)
 smaxIndexPrim :: (PrimElt a, NumElt a) => Shaped sh a -> IIxS sh
 smaxIndexPrim (Shaped arr) = ixsFromIxX (mmaxIndexPrim arr)
 
+{-# INLINEABLE sdot1Inner #-}
 sdot1Inner :: forall sh n a. (PrimElt a, NumElt a)
            => Proxy n -> Shaped (sh ++ '[n]) a -> Shaped (sh ++ '[n]) a -> Shaped sh a
 sdot1Inner Proxy sarr1@(Shaped arr1) (Shaped arr2)
@@ -272,6 +281,7 @@ sdot1Inner Proxy sarr1@(Shaped arr1) (Shaped arr2)
         -> Shaped (mdot1Inner (Proxy @(Just n)) arr1 arr2)
       _ -> error "unreachable"
 
+{-# INLINE sdot #-}
 -- | This has a temporary, suboptimal implementation in terms of 'mflatten'.
 -- Prefer 'sdot1Inner' if applicable.
 sdot :: (PrimElt a, NumElt a) => Shaped sh a -> Shaped sh a -> a
-- 
cgit v1.2.3-70-g09d2