tmp

Author: ana-pantilie

plutus-core/cost-model/budgeting-bench/Benchmarks/Values.hs

@@ -296,15 +296,10 @@ generateConstrainedValue numPolicies tokensPerPolicy g = do
   policyIds <- replicateM numPolicies (generateKey g)
   tokenNames <- replicateM tokensPerPolicy (generateKey g)
 
-  -- Generate positive quantities (1 to 1000000)
-  let quantity :: Int -> Int -> Integer
-      quantity policyIndex tokenIndex =
-        fromIntegral (1 + (policyIndex * 1_000 + tokenIndex) `mod` 1_000_000)
-
-      nestedMap :: [(K, [(K, Integer)])]
+  let nestedMap :: [(K, [(K, Integer)])]
       nestedMap =
         [ ( policyId
-          , [ (tokenName, quantity policyIndex tokenIndex)
+          , [ (tokenName, genQuantity policyIndex tokenIndex)
             | (tokenIndex, tokenName) <- zip [0 ..] tokenNames
             ]
           )
@@ -347,3 +342,8 @@ generateKey g = do
 -- | Generate random key as ByteString (for lookup arguments)
 generateKeyBS :: (StatefulGen g m) => g -> m ByteString
 generateKeyBS = uniformByteStringM Value.maxKeyLen
+
+-- | Generate positive quantities (1 to 1000000)
+genQuantity :: Int -> Int -> Integer
+genQuantity policyIndex tokenIndex =
+  fromIntegral (1 + (policyIndex * 1_000 + tokenIndex) `mod` 1_000_000)

plutus-core/cost-model/budgeting-bench/Common.hs

@@ -431,3 +431,36 @@ createThreeTermBuiltinBenchWithWrappers (wrapX, wrapY, wrapZ) fun tys xs ys zs =
       [mkBM x y z | z <- zs] | y <- ys] | x <- xs]
   where mkBM x y z = benchDefault (showMemoryUsage (wrapZ z)) $ mkApp3 fun tys x y z
 
+{- See Note [Adjusting the memory usage of arguments of costing benchmarks]. -}
+createFourTermBuiltinBenchElementwiseWithWrappers
+  :: ( fun ~ DefaultFun
+     , uni ~ DefaultUni
+     , uni `HasTermLevel` a
+     , uni `HasTermLevel` b
+     , uni `HasTermLevel` c
+     , uni `HasTermLevel` d
+     , ExMemoryUsage a'
+     , ExMemoryUsage b'
+     , ExMemoryUsage c'
+     , ExMemoryUsage d'
+     , NFData a
+     , NFData b
+     , NFData c
+     , NFData d
+     )
+  => (a -> a', b -> b', c -> c', d -> d')
+  -> fun
+  -> [Type tyname uni ()]
+  -> [(a,b,c,d)]
+  -> Benchmark
+createFourTermBuiltinBenchElementwiseWithWrappers (wrapW, wrapX, wrapY, wrapZ) fun tys inputs =
+  bgroup (show fun) $
+  fmap
+  (\(w, x, y, z) ->
+     bgroup (showMemoryUsage $ wrapW w)
+     [bgroup (showMemoryUsage $ wrapX x)
+      [bgroup (showMemoryUsage $ wrapY y) [mkBM w x y z]]
+     ]
+  )
+  inputs
+  where mkBM w x y z = benchDefault (showMemoryUsage $ wrapZ z) $ mkApp4 fun tys w x y z