Test gmm

1 files changed, 6 insertions, 3 deletions

diff --git a/src/ForwardAD.hs b/src/ForwardAD.hs
index 67d22dd..b95385c 100644
--- a/src/ForwardAD.hs
+++ b/src/ForwardAD.hs
@@ -7,11 +7,12 @@
 module ForwardAD where
 
 import Data.Bifunctor (bimap)
--- import Data.Foldable (toList)
+
+-- import Debug.Trace
+-- import AST.Pretty
 
 import Array
 import AST
--- import AST.Bindings
 import Data
 import ForwardAD.DualNumbers
 import Interpreter
@@ -214,6 +215,8 @@ dnOnehotEnvs (t `SCons` env) (Value x `SCons` val) =
 drevByFwd :: SList STy env -> Ex env t -> SList Value env -> Rep (Tan t) -> SList Value (TanE env)
 drevByFwd env expr input dres =
   let outty = typeOf expr
-  in dnOnehotEnvs env input $ \dnInput ->
+  in -- trace ("fwd: running: " ++ ppExpr (dne env) (dfwdDN expr)) $
+     dnOnehotEnvs env input $ \dnInput ->
+       -- trace (showEnv (dne env) dnInput) $
        let (_, outtan) = unzipDN outty (interpretOpen False dnInput (dfwdDN expr))
        in dotprodTan outty outtan dres