Add guessPolynomialComplexity

Bodigrim · Jul 6, 2024 · 96f734c · 96f734c
1 parent b6a1870
commit 96f734c
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,7 @@
 ## 0.1.1
 
 * Introduce `Test.Tasty.Bench.Crossover`.
+* Introduce `guessPolynomialComplexity`.
 
 ## 0.1
 

diff --git a/src/Test/Tasty/Bench/Fit.hs b/src/Test/Tasty/Bench/Fit.hs
@@ -1,6 +1,8 @@
+{-# HLINT ignore "Avoid restricted function" #-}
 {-# LANGUAGE NumDecimals #-}
 {-# LANGUAGE RecordWildCards #-}
 {-# LANGUAGE ScopedTypeVariables #-}
+{-# OPTIONS_GHC -Wno-unrecognised-pragmas #-}
 
 -- | Guess complexity of the function.
 module Test.Tasty.Bench.Fit (
@@ -14,6 +16,7 @@ module Test.Tasty.Bench.Fit (
   Complexity (..),
   Measurement (..),
   guessComplexity,
+  guessPolynomialComplexity,
   evalComplexity,
 
   -- * Predicates
@@ -39,6 +42,7 @@ import Test.Tasty.Bench.Fit.Complexity (
   Complexity (..),
   evalComplexity,
   guessComplexity,
+  guessPolynomialComplexity,
   isConstant,
   isCubic,
   isLinear,

diff --git a/src/Test/Tasty/Bench/Fit/Complexity.hs b/src/Test/Tasty/Bench/Fit/Complexity.hs
@@ -9,6 +9,7 @@
 module Test.Tasty.Bench.Fit.Complexity (
   Complexity (..),
   guessComplexity,
+  guessPolynomialComplexity,
   evalComplexity,
 
   -- * Predicates
@@ -179,6 +180,19 @@ guessComplexity xys =
       | otherwise =
           [c1, c2]
 
+-- | Same as guessComplexity, but the power of \( \log x \) ('cmplLogPower')
+-- is pinned to be 0.
+--
+-- @since 0.1.1
+guessPolynomialComplexity :: Map Word Measurement -> Complexity
+guessPolynomialComplexity xys =
+  trace'
+    ("guessPolynomialComplexity " ++ show (M.assocs xys))
+    bestOf
+    [cmpl1, cmpl2]
+  where
+    (cmpl1, cmpl2) = guessComplexityForFixedLog xys 0
+
 guessComplexityForFixedLog
   :: Map Word Measurement
   -> Word