diff options
Diffstat (limited to 'src/CHAD.hs')
| -rw-r--r-- | src/CHAD.hs | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/CHAD.hs b/src/CHAD.hs index d7d7da2..a5a5719 100644 --- a/src/CHAD.hs +++ b/src/CHAD.hs @@ -896,9 +896,10 @@ drev des = \case subtape (EReplicate1Inner ext en1 e1) sub - (ELet ext (EFold1Inner ext (EPlus ext eltty (EVar ext (d2 eltty) (IS IZ)) (EVar ext (d2 eltty) IZ)) - (EZero ext eltty) - (EVar ext (STArr (SS ndim) (d2 eltty)) IZ)) $ + (ELet ext (EFold1Inner ext Commut + (EPlus ext eltty (EVar ext (d2 eltty) (IS IZ)) (EVar ext (d2 eltty) IZ)) + (EZero ext eltty) + (EVar ext (STArr (SS ndim) (d2 eltty)) IZ)) $ weakenExpr (WCopy WSink) e2) EIdx0 _ e |