diff options
| -rw-r--r-- | OneShotTest.hs | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/OneShotTest.hs b/OneShotTest.hs new file mode 100644 index 0000000000..b595285069 --- /dev/null +++ b/OneShotTest.hs @@ -0,0 +1,22 @@ +{-# OPTIONS_GHC -fno-call-arity #-} + +module OneShotTest (foldlB, foldlA, fooA, fooB, fooC) where + +import GHC.Prim (oneShot) + +foldlA, foldlB :: (x -> a -> a) -> a -> [x] -> a + +foldlA k a xs = foldr (\v f a -> f (v `k` a)) id xs a +{-# INLINEABLE foldlA #-} + +foldlB k a xs = foldr (\v f -> oneShot (\ a -> f (v `k` a))) id xs a +{-# INLINEABLE foldlB #-} + +f :: Int -> Bool +f 0 = True +f 1 = False +{-# NOINLINE f #-} + +fooA = foldlA (+) 0 . filter f +fooB = foldlB (+) 0 . filter f +fooC = foldl (+) 0 . filter f |