diff options
| author | Tom Smeding <t.j.smeding@uu.nl> | 2025-03-20 18:32:22 +0100 |
|---|---|---|
| committer | Tom Smeding <t.j.smeding@uu.nl> | 2025-03-20 18:32:22 +0100 |
| commit | d030802dd6d960afa80ac84a5580a46d39c02822 (patch) | |
| tree | 0c40e8eea6fe12cab0bd74e5e4f457e13bbf9afd /src/CHAD.hs | |
| parent | 146a846f799f63cd98eee2149c417686adba17a9 (diff) | |
Commutativity marker on fold1i
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 |