summaryrefslogtreecommitdiff
path: root/src/ForwardAD.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/ForwardAD.hs')
-rw-r--r--src/ForwardAD.hs9
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