diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2025-10-16 15:44:20 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2025-10-16 15:44:20 +0200 |
| commit | 15d586515dea0b352be8b71ea9bade3ae2389784 (patch) | |
| tree | ce9e0cd1e02fa8a0ee12887ec770be20857a6898 | |
| parent | 3dbf6044a2ddfa607eb52f0e41dc17602119d7e2 (diff) | |
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!
| -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 |