From cbe6472a14cc0887295034bb29546dd1a1f083fd Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Fri, 8 Nov 2024 22:17:56 +0100
Subject: WIP maximum/minimum

---
 src/ForwardAD/DualNumbers.hs | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/ForwardAD/DualNumbers.hs')

diff --git a/src/ForwardAD/DualNumbers.hs b/src/ForwardAD/DualNumbers.hs
index f2ded6e..056fcb3 100644
--- a/src/ForwardAD/DualNumbers.hs
+++ b/src/ForwardAD/DualNumbers.hs
@@ -131,6 +131,13 @@ dfwdDN = \case
          (ESum1Inner ext (dfwdDN e))
   EUnit _ e -> EUnit ext (dfwdDN e)
   EReplicate1Inner _ a b -> EReplicate1Inner ext (dfwdDN a) (dfwdDN b)
+  EMaximum1Inner _ e ->
+    let STArr n (STScal t) = typeOf e
+    in scalTyCase t
+         -- TODO: do roughly the same as what CHAD does, but forward
+         (_ (dfwdDN e))
+         _
+  EMinimum1Inner _ e -> EMinimum1Inner ext (dfwdDN e)
   EConst _ t x -> scalTyCase t
     (EPair ext (EConst ext t x) (EConst ext t 0.0))
     (EConst ext t x)
-- 
cgit v1.2.3-70-g09d2