From 15d586515dea0b352be8b71ea9bade3ae2389784 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Thu, 16 Oct 2025 15:44:20 +0200
Subject: Add pruneExpr to benchmarks
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

       neural/default neural/accum gmm/default gmm/accum
before    157.1 μs      117.9 μs    2.087 ms    2.040 ms
after     106.6 μs      107.0 μs    1.931 ms    1.780 ms

I call that a win!
---
 bench/Main.hs | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'bench')

diff --git a/bench/Main.hs b/bench/Main.hs
index 358ba31..53657b4 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -17,6 +17,7 @@ import Data.Kind (Constraint)
 import GHC.Exts (withDict)
 
 import AST
+import AST.Count
 import AST.UnMonoid
 import Array
 import qualified CHAD (defaultConfig)
@@ -34,6 +35,7 @@ import Simplify
 gradCHAD :: KnownEnv env => CHADConfig -> Ex env (TScal TF64) -> IO (SList Value env -> IO (Double, Rep (Tup (D2E env))))
 gradCHAD config term =
   compile knownEnv $
+    simplifyFix $ pruneExpr knownEnv $
     simplifyFix $ unMonoid $ simplifyFix $
       ELet ext (EConst ext STF64 1.0) $ chad' config knownEnv term
 
-- 
cgit v1.2.3-70-g09d2