EnzymeAD / EnzymeAD/Enzyme-JAX
broadcast(convert(...)) to convert(broadcast()) based on types
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
```
%19 = stablehlo.convert %arg11 : (tensor<32xbf16>) -> tensor<32xf32>
%20 = stablehlo.broadcast_in_dim %19, dims = [2] : (tensor<32xf32>) -> tensor<128x12x32xf32>
```
We should broadcast and then convert when increasing precision
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by locating the code that emits the StableHLO convert and broadcast_in_dim operations shown in the issue. Trace how operand and result types determine their ordering, then verify that increasing precision broadcasts before converting and that the relevant regression coverage passes.
Written by the indexing model from the issue text.
Assessment
- Domain
- compilers
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 45/100