commit b08f43071a0c7ebae505f2cfb060fc01b9eade33
author Julian Walker <julianreidwalker@gmail.com> Thu Feb 29 11:05:15 2024 -0800
committer GitHub <noreply@github.com> Thu Feb 29 11:05:15 2024 -0800
tree f93934d95ff4587a9b74f113f7bbbfb1e333d2d1
parent 890b070bd94a62dc90203ed0514b212836af46ac

Bump StableHLO to 0264c4d64c82ae74a54b85d274eec5084c2c0abf (#16561)

- This update spans a change made to StableHLO to switch op attributes
from `I64ElementsAttr` to `DenseI64ArrayAttr`, discussed here:
https://groups.google.com/a/openxla.org/g/openxla-discuss/c/hEoA4V5DZF0/m/rdNEiM20BgAJ
- Unfortunately, StableHLO generated prior to this change upstream may
not be compatible with the StableHLO input plug-in in IREE after this
PR.
- Min version of jax/jaxlib compatible with this change: v0.4.24
- Min version of TF compatible with this change: v2.16.0
- Some of the changes here were propagated from the subset of StableHLO
to LinAlg lowers that live in the StableHLO repro.

To update stored SHLO bytecode, I did the following:

1. Compiled openxla/stablehlo with current IREE repo version and
upgraded IREE repo versions to `old_stablehlo` and `new_stablehlo`
respectively.
1. `old_stablehlo/build/bin/stablehlo-opt --mlir-print-op-generic
<model>.mlirbc > <model>.mlir`
1. Manually edit <model>.mlir to bring it up to new spec
1. `new_stablehlo/build/bin/stablehlo-opt --emit-bytecode <model>.mlir >
new_<model>.mlirbc`

The following covered most of the edits that needed to be made to the
model MLIR:
```
sed -i \
-e 's/dimensions = dense<\([0-9]*\)> : tensor<1xi64>/dimensions = array<i64: \1>/g' \
-e 's/dimensions = dense<\[\([0-9]*\), \([0-9]*\)\]> : tensor<2xi64>/dimensions = array<i64: \1, \2>/g' \
-e 's/rhs_dilation = dense<\([0-9]*\)> : tensor<2xi64>/rhs_dilation = array<i64: \1, \1>/g' \
-e 's/window_strides = dense<\([0-9]*\)> : tensor<2xi64>/window_strides = array<i64: \1, \1>/g' \
model.mlir
```

ci-extra: build_e2e_test_artifacts, test_benchmark_suites