EnzymeAD / EnzymeAD/Enzyme-JAX

GB Stablehlo optimization

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

Description

```
// CHECK-NEXT: %0 = stablehlo.iota dim = 0 : tensor<185xi64>
// CHECK-NEXT: %1 = stablehlo.add %0, %c_1 : tensor<185xi64>
// CHECK-NEXT: %2 = stablehlo.convert %1 : (tensor<185xi64>) -> tensor<185xf64>
// CHECK-NEXT: %3 = stablehlo.abs %2 : tensor<185xf64>
```

we shouldn't need to do the abs here as iota >= 0, as is iota + 1

Contributor guide

No contributing guide indexed for this repository

Research direction

The issue only shows a StableHLO CHECK-NEXT sequence; begin by locating the test or lowering pass that produces it and inspect the iota/add/convert/abs pattern. Done means the redundant abs is absent from the generated sequence and the relevant check passes.

Written by the indexing model from the issue text.

Assessment

Domain
compilers
Issue type
Refactor
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.