From 2dd47698f186a7886e9df98f41905d5c2ebefe63 Mon Sep 17 00:00:00 2001
From: Alexey Kuleshevich <alexey@kuleshevi.ch>
Date: Sat, 14 Dec 2024 10:10:40 -0700
Subject: [PATCH] Make benchmark compatible with older primitive

This is necessary to allow testing with lts-9 snapshot, since stack is
still capable of installing ghc-8.0 and ghc-8.2, while ghcup is having
issues in achieving that on CI
---
 bench/Main.hs     | 16 ++++++++++++----
 random.cabal      |  2 +-
 stack-custom.yaml |  1 -
 3 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/bench/Main.hs b/bench/Main.hs
index cb0c5ba0..d06e2a95 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -1,4 +1,5 @@
 {-# LANGUAGE BangPatterns #-}
+{-# LANGUAGE CPP #-}
 {-# LANGUAGE FlexibleContexts #-}
 {-# LANGUAGE ScopedTypeVariables #-}
 module Main (main) where
@@ -13,9 +14,11 @@ import Foreign.C.Types
 import Numeric.Natural (Natural)
 import System.Random.SplitMix as SM
 import Test.Tasty.Bench
+#if MIN_VERSION_primitive(0,7,1)
 import Control.Monad.Primitive
-import Data.Primitive.PrimArray
 import Data.Primitive.Types
+import Data.Primitive.PrimArray
+#endif
 
 import System.Random.Stateful
 
@@ -200,7 +203,9 @@ main = do
             in pureUniformRBench (Proxy :: Proxy Natural) range sz
           ]
         , bgroup "floating"
-          [ bgroup "IO"
+          [
+#if MIN_VERSION_primitive(0,7,1)
+            bgroup "IO"
             [ env ((,) <$> getStdGen <*> newAlignedPinnedPrimArray sz) $ \ ~(gen, ma) ->
                 bench "uniformFloat01M" $
                 nfIO (runStateGenT gen (fillMutablePrimArrayM uniformFloat01M ma))
@@ -214,7 +219,9 @@ main = do
                 bench "uniformDoublePositive01M" $
                 nfIO (runStateGenT gen (fillMutablePrimArrayM uniformDoublePositive01M ma))
             ]
-          , bgroup "State"
+          ,
+#endif
+            bgroup "State"
             [ env getStdGen $
                 bench "uniformFloat01M" . nf (`runStateGen` (replicateM_ sz . uniformFloat01M))
             , env getStdGen $
@@ -329,7 +336,7 @@ genMany f g0 n = go 0 $ f g0
       | i < n = go (i + 1) $ f g
       | otherwise = y
 
-
+#if MIN_VERSION_primitive(0,7,1)
 fillMutablePrimArrayM ::
      (Prim a, PrimMonad m)
   => (gen -> m a)
@@ -343,3 +350,4 @@ fillMutablePrimArrayM f ma g = do
         | otherwise = pure ()
   go 0
   unsafeFreezePrimArray ma
+#endif
diff --git a/random.cabal b/random.cabal
index 891c63d5..69756e7e 100644
--- a/random.cabal
+++ b/random.cabal
@@ -189,7 +189,7 @@ benchmark bench
     build-depends:
         base,
         mtl,
-        primitive >= 0.7.1,
+        primitive,
         random,
         splitmix >=0.1 && <0.2,
         tasty-bench
diff --git a/stack-custom.yaml b/stack-custom.yaml
index 37e69f0d..e7f745cb 100644
--- a/stack-custom.yaml
+++ b/stack-custom.yaml
@@ -13,7 +13,6 @@ extra-deps:
 - tasty-bench-0.2.3@sha256:daa2221a1b1c65990633a51236f1cb4a52cba8ef0f0731f653e712a8bab07616,1319
 - inspection-testing-0.4.5.0@sha256:938e7ce2ef42033071a5e60198c6e19ab61c411f5879b85821247a504f131768,8058
 - tasty-inspection-testing-0.1@sha256:9c5e76345168fd3a59b43d305eebf8df3c792ce324c66bbdee45b54aa7d2c0ad,1214
-- primitive-0.7.4.0@sha256:89b88a3e08493b7727fa4089b0692bfbdf7e1e666ef54635f458644eb8358764,2857
 - vector-0.12.3.1@sha256:fffbd00912d69ed7be9bc7eeb09f4f475e0d243ec43f916a9fd5bbd219ce7f3e,8238
 - data-array-byte-0.1.0.1@sha256:ad89e28b2b046175698fbf542af2ce43e5d2af50aae9f48d12566b1bb3de1d3c,1989
 - optparse-applicative-0.14.2.0@sha256:cfbc2df0d9e144d343ccea6d25ab27543c15ea17d2abcbabcdf76030a2236383,4359