From c50aecbb209bb99da074b6c6911b672d0f1d1388 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Tue, 21 Oct 2025 22:45:17 +0200
Subject: bench: Simplify before differentiating

---
 bench/Main.hs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'bench/Main.hs')

diff --git a/bench/Main.hs b/bench/Main.hs
index 53657b4..34e8bae 100644
--- a/bench/Main.hs
+++ b/bench/Main.hs
@@ -36,8 +36,11 @@ gradCHAD :: KnownEnv env => CHADConfig -> Ex env (TScal TF64) -> IO (SList Value
 gradCHAD config term =
   compile knownEnv $
     simplifyFix $ pruneExpr knownEnv $
-    simplifyFix $ unMonoid $ simplifyFix $
-      ELet ext (EConst ext STF64 1.0) $ chad' config knownEnv term
+    simplifyFix $ unMonoid $
+    simplifyFix $
+    ELet ext (EConst ext STF64 1.0) $
+      chad' config knownEnv $
+      simplifyFix term
 
 type AllNFDataRep :: [Ty] -> Constraint
 type family AllNFDataRep env where
-- 
cgit v1.2.3-70-g09d2