diff options
| -rw-r--r-- | bench/Main.hs | 2 |
1 files changed, 2 insertions, 0 deletions
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 |