From b8c162ce9cb1faeec621b751fff9aff46e022417 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Thu, 14 Nov 2024 19:27:57 +0100
Subject: Configuration for CHAD

---
 test/Main.hs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'test')

diff --git a/test/Main.hs b/test/Main.hs
index d617228..7cb15d5 100644
--- a/test/Main.hs
+++ b/test/Main.hs
@@ -24,6 +24,7 @@ import Hedgehog.Main
 import Array
 import AST
 import AST.Pretty
+import CHAD (defaultConfig)
 import CHAD.Top
 import CHAD.Types
 import qualified Example
@@ -41,7 +42,7 @@ data SimplIters = SimplIters Int | SimplFix
 -- In addition to the gradient, also returns the pretty-printed differentiated term.
 gradientByCHAD :: forall env. SimplIters -> SList STy env -> Ex env (TScal TF64) -> SList Value env -> (String, (Double, SList Value (D2E env)))
 gradientByCHAD = \simplIters env term input ->
-  let dtermNonSimpl = ELet ext (EConst ext STF64 1.0) $ chad' env term
+  let dtermNonSimpl = ELet ext (EConst ext STF64 1.0) $ chad' defaultConfig env term
       dterm | Dict <- envKnown env
             = case simplIters of
                 SimplIters n -> simplifyN n dtermNonSimpl
-- 
cgit v1.2.3-70-g09d2