EnzymeAD / EnzymeAD/Enzyme-JAX

Concat of operation with slices on boundary -> dynamic_slice

Open
#630 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

```
%240 = ...
%242 = stablehlo.slice %240 [0:1, 0:1, 0:2032] : (tensor<1x2032x2032xf64>) -> tensor<1x1x2032xf64> loc(#loc1167)
%243 = stablehlo.slice %240 [0:1, 2031:2032, 0:2032] : (tensor<1x2032x2032xf64>) -> tensor<1x1x2032xf64> loc(#loc1168)
%244 = stablehlo.concatenate %242, %240, %243, dim = 1 : (tensor<1x1x2032xf64>, tensor<1x2032x2032xf64>, tensor<1x1x2032xf64>) -> tensor<1x2034x2032xf64> loc(#loc1169)
```

->
reduce window with padding 1_1 and dilation 2

x/ref https://github.com/openxla/stablehlo/blob/c2fccec9f02daa2d709001ed5f37924d2526676f/stablehlo/dialect/StablehloOps.td#L2847 and https://github.com/openxla/stablehlo/blob/c2fccec9f02daa2d709001ed5f37924d2526676f/stablehlo/dialect/StablehloOps.td#L2858

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.