EnzymeAD / EnzymeAD/Enzyme-JAX

slice of single use rotate -> rotate slice

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

```

%1258 = stablehlo.slice %1255 [8:12, 6:1526, 0:3056] : (tensor<20x1536x3056xf64>) -> tensor<4x1520x3056xf64> loc(#loc4911)

%2998 = "enzymexla.rotate"(%1258) <{amount = 1 : si32, dimension = 2 : si32}> : (tensor<4x1520x3056xf64>) -> tensor<4x1520x3056xf64> loc(#loc4095)
%3366 = stablehlo.slice %2998 [0:4, 1:1520, 0:3056] : (tensor<4x1520x3056xf64>) -> tensor<4x1519x3056xf64> loc(#loc3026)
```

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by tracing the stablehlo.slice and enzymexla.rotate operations shown in the issue through the Enzyme-JAX compiler pipeline. Determine where the single-use pattern is handled and verify the intended result by checking the generated IR: the slice should be moved across the rotate as described by the title.

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
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.