EnzymeAD / EnzymeAD/Enzyme-JAX
More wrap rotate fixups
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
```
%685 = stablehlo.slice %arg20 [8:12, 8:6136, 8:12280] : (tensor<20x6144x12288xf32>) -> tensor<4x6128x12272xf32> loc(#loc4553)
%1235 = "enzymexla.rotate"(%685) <{amount = 2 : si32, dimension = 2 : si32}> : (tensor<4x6128x12272xf32>) -> tensor<4x6128x12272xf32> loc(#loc3774)
%1635 = stablehlo.slice %arg20 [8:12, 8:6136, 10:11] : (tensor<20x6144x12288xf32>) -> tensor<4x6128x1xf32> loc(#loc2576)
%1636 = stablehlo.concatenate %1235, %1635, dim = 2 : (tensor<4x6128x12272xf32>, tensor<4x6128x1xf32>) -> tensor<4x6128x12273xf32> loc(#loc2576)
```
```
%22 = stablehlo.slice %15 [8:12, 8:6136, 12269:12270] : (tensor<20x6144x12272xf32>) -> tensor<4x6128x1xf32> loc(#loc3742)
%23 = stablehlo.slice %15 [8:12, 8:6136, 0:12272] : (tensor<20x6144x12272xf32>) -> tensor<4x6128x12272xf32> loc(#loc3742)
%648 = "enzymexla.rotate"(%23) <{amount = 12270 : si32, dimension = 2 : si32}> : (tensor<4x6128x12272xf32>) -> tensor<4x6128x12272xf32> loc(#loc3742)
%1847 = stablehlo.concatenate %22, %648, dim = 2 : (tensor<4x6128x1xf32>, tensor<4x6128x12272xf32>) -> tensor<4x6128x12273xf32> loc(#loc3603)
```
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by tracing the wrap and rotate handling for the shown StableHLO slices, enzymexla.rotate operations, and concatenates. Compare both examples to identify the missing fixup behavior; done means the affected rotate cases produce the intended concatenated shapes and have regression coverage.
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
- Needs clarification
- Newbie friendliness
- 25/100