PennyLaneAI / PennyLaneAI/catalyst
bufferization option copy-before-write may not be compatible with Enzyme
Open
Nobody has claimed this yet.
bug
compiler
- Dominant language
- Python
- Stars
- 234
- Forks
- 84
- Avg merge
- 2d 15h
- Merged PRs (30d)
- 66
Description
Looking into speeding up bufferization I attempted to avoid the bufferization analysis by just increasing memory usage. This lead me to enabling the bufferization option copy-before-write for all workloads and running the tests. Some gradient python tests failed with this option.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the reported failures in the gradient Python tests with the bufferization option copy-before-write enabled for all workloads. Read the Enzyme integration and bufferization handling around those failures, then confirm that the affected tests pass without breaking the intended bufferization speedup.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- compilers
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 35/100