diff options
Diffstat (limited to 'src/CHAD/Drev.hs')
| -rw-r--r-- | src/CHAD/Drev.hs | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/CHAD/Drev.hs b/src/CHAD/Drev.hs index bfa964b..9f2921c 100644 --- a/src/CHAD/Drev.hs +++ b/src/CHAD/Drev.hs @@ -1230,7 +1230,8 @@ drev des accumMap sd = \case sub (ELet ext (EFold1Inner ext Commut (let t = STPair (applySparse sdElt' (d2 eltty)) (applySparse sdElt' (d2 eltty)) - in sparsePlus (d2M eltty) sdElt' (EFst ext (EVar ext t IZ)) (ESnd ext (EVar ext t IZ))) + in EPlus ext (applySparse sdElt' (d2M eltty)) + (EFst ext (EVar ext t IZ)) (ESnd ext (EVar ext t IZ))) (inj2 (ENil ext)) (emap (inj1 (evar IZ)) $ EVar ext (STArr (SS ndim) (applySparse sdElt (d2 eltty))) IZ)) $ weakenExpr (WCopy WSink) e2) |