EnzymeAD / EnzymeAD/Enzyme-JAX
More aggressive slice of update_without_corners
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
```
%338 = "enzymexla.update_without_corners"(%79, %336) <{dimensionX = 0 : i64, dimensionY = 1 : i64, x1 = 1 : i64, x2 = 5 : i64, y1 = 1 : i64, y2 = 6129 : i64}> : (tensor<6x6130x12272xf64>, tensor<6x6130x12272xf64>) -> tensor<6x6130x12272xf64>
%339 = "enzymexla.update_without_corners"(%80, %337) <{dimensionX = 0 : i64, dimensionY = 1 : i64, x1 = 1 : i64, x2 = 5 : i64, y1 = 1 : i64, y2 = 6129 : i64}> : (tensor<6x6130x12272xf64>, tensor<6x6130x12272xf64>) -> tensor<6x6130x12272xf64>
%389 = stablehlo.slice %338 [1:5, 1:6129, 0:12272] : (tensor<6x6130x12272xf64>) -> tensor<4x6128x12272xf64>
%1288 = stablehlo.slice %338 [5:6, 0:1, 0:12272] : (tensor<6x6130x12272xf64>) -> tensor<1x1x12272xf64>
%1289 = stablehlo.slice %338 [5:6, 6129:6130, 0:12272] : (tensor<6x6130x12272xf64>) -> tensor<1x1x12272xf64>
%394 = stablehlo.slice %339 [1:5, 1:6129, 0:12272] : (tensor<6x6130x12272xf64>) -> tensor<4x6128x12272xf64>
%1296 = stablehlo.slice %339 [5:6, 0:1, 0:12272] : (tensor<6x6130x12272xf64>) -> tensor<1x1x12272xf64>
%1297 = stablehlo.slice %339 [5:6, 6129:6130, 0:12272] : (tensor<6x6130x12272xf64>) -> tensor<1x1x12272xf64>
```
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.