EnzymeAD / EnzymeAD/Enzyme-JAX
Concat of many same argument -> broadcast
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
Running this code:
https://github.com/PRONTOLab/GB-25/blob/b5a87acdbcd7fc41143edecbc7cf0371f65d7d2f/correctness/correctness_baroclinic_instability_simulation_run.jl
With this branch of Oceananigans.jl: `jlk9/stack-spacings`
Produces this MLIR output as the code hangs and fails to compile:
[module_120_dYTv_pre_all_pm.mlir.txt](https://github.com/user-attachments/files/21550016/module_120_dYTv_pre_all_pm.mlir.txt)
@wsmoses
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by running the linked correctness_baroclinic_instability_simulation_run.jl reproduction from the specified Oceananigans.jl branch and inspect the attached MLIR output. Compare the behavior with the merged pull request #1228; done means the many-same-argument concat no longer hangs or fails to compile.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- julia
- Domain
- compilers
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100