EnzymeAD / EnzymeAD/Enzyme-JAX

broadcast(convert(...)) to convert(broadcast()) based on types

Open
#1,350 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

```
%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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.