EnzymeAD / EnzymeAD/Enzyme-JAX

Rework ReshapeBroadcast passes to reuse Helper functions

Open
#1,090 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
MLIR
Stars
131
Forks
53
Avg merge
1d 10h
Merged PRs (30d)
193

Description

https://github.com/EnzymeAD/Enzyme-JAX/blob/7277ac177c3547892a91d2397f8f21ddfbb3d1a2/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp#L614

Contributor guide

No contributing guide indexed for this repository

Research direction

Start at src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp around line 614 and inspect the ReshapeBroadcast passes and the available Helper functions. Rework those passes to reuse the helpers, then verify that the pass behavior remains correct.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp
Domain
compilers
Issue type
Refactor
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
45/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.