EnzymeAD / EnzymeAD/Enzyme

MLIR support for differentiating parallel loops

Open
#2,658 2 comments 0 reactions 0 assignees View on GitHub
mlir
Dominant language
LLVM
Stars
1.7k
Forks
188
Avg merge
1d 22h
Merged PRs (30d)
26

Description

We've noticed some support for differentiating parallel loops was recently added (e.g., 902bf1f for `scf.parallel`), but it seems incomplete. We were wondering if there is work to more fully support differentiating parallel loops (in particular, forward mode and `scf.parallel.reduce`), either planned or on-going? Differentiating parallel loops is central to our work, which is something we could contribute to, but don't want to duplicate efforts.

@brian-kelley

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.