summaryrefslogtreecommitdiff
path: root/src/ForwardAD
diff options
context:
space:
mode:
authorTom Smeding <tom@tomsmeding.com>2024-10-22 22:02:06 +0200
committerTom Smeding <tom@tomsmeding.com>2024-10-22 22:02:06 +0200
commit79e072eddf0ec2a97ca455c27cb5ff6f2132bbab (patch)
tree2099dc7e9d9a1109d844bca73277ca82983a02c2 /src/ForwardAD
parente7d7ac0fd8b81c1d6fae9ab7c1e4654133c631ea (diff)
Differentiate Replicate
Diffstat (limited to 'src/ForwardAD')
-rw-r--r--src/ForwardAD/DualNumbers.hs2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/ForwardAD/DualNumbers.hs b/src/ForwardAD/DualNumbers.hs
index 3e45ce7..a93b8e6 100644
--- a/src/ForwardAD/DualNumbers.hs
+++ b/src/ForwardAD/DualNumbers.hs
@@ -153,7 +153,7 @@ dfwdDN = \case
EBuild1 _ a b -> EBuild1 ext (dfwdDN a) (dfwdDN b)
EBuild _ n a b
| Refl <- dnPreservesTupIx n -> EBuild ext n (dfwdDN a) (dfwdDN b)
- EFold1Inner _ a b -> EFold1Inner ext (dfwdDN a) (dfwdDN b)
+ EFold1Inner _ a b c -> EFold1Inner ext (dfwdDN a) (dfwdDN b) (dfwdDN c)
ESum1Inner _ e ->
let STArr n (STScal t) = typeOf e
pairty = (STPair (STScal t) (STScal t))