From a46f53695d1dfab8834c7cc52707c0c0bb9b8ba0 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Sun, 10 Nov 2024 22:40:54 +0100
Subject: Test gmm

---
 src/ForwardAD.hs | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'src/ForwardAD.hs')

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
-- 
cgit v1.2.3-70-g09d2