)]}' { "commit": "b08f43071a0c7ebae505f2cfb060fc01b9eade33", "tree": "f93934d95ff4587a9b74f113f7bbbfb1e333d2d1", "parents": [ "890b070bd94a62dc90203ed0514b212836af46ac" ], "author": { "name": "Julian Walker", "email": "julianreidwalker@gmail.com", "time": "Thu Feb 29 11:05:15 2024 -0800" }, "committer": { "name": "GitHub", "email": "noreply@github.com", "time": "Thu Feb 29 11:05:15 2024 -0800" }, "message": "Bump StableHLO to 0264c4d64c82ae74a54b85d274eec5084c2c0abf (#16561)\n\n- This update spans a change made to StableHLO to switch op attributes\r\nfrom `I64ElementsAttr` to `DenseI64ArrayAttr`, discussed here:\r\nhttps://groups.google.com/a/openxla.org/g/openxla-discuss/c/hEoA4V5DZF0/m/rdNEiM20BgAJ\r\n- Unfortunately, StableHLO generated prior to this change upstream may\r\nnot be compatible with the StableHLO input plug-in in IREE after this\r\nPR.\r\n- Min version of jax/jaxlib compatible with this change: v0.4.24\r\n- Min version of TF compatible with this change: v2.16.0\r\n- Some of the changes here were propagated from the subset of StableHLO\r\nto LinAlg lowers that live in the StableHLO repro.\r\n\r\nTo update stored SHLO bytecode, I did the following:\r\n\r\n1. Compiled openxla/stablehlo with current IREE repo version and\r\nupgraded IREE repo versions to `old_stablehlo` and `new_stablehlo`\r\nrespectively.\r\n1. `old_stablehlo/build/bin/stablehlo-opt --mlir-print-op-generic\r\n\u003cmodel\u003e.mlirbc \u003e \u003cmodel\u003e.mlir`\r\n1. Manually edit \u003cmodel\u003e.mlir to bring it up to new spec\r\n1. `new_stablehlo/build/bin/stablehlo-opt --emit-bytecode \u003cmodel\u003e.mlir \u003e\r\nnew_\u003cmodel\u003e.mlirbc`\r\n\r\nThe following covered most of the edits that needed to be made to the\r\nmodel MLIR:\r\n```\r\nsed -i \\\r\n-e \u0027s/dimensions \u003d dense\u003c\\([0-9]*\\)\u003e : tensor\u003c1xi64\u003e/dimensions \u003d array\u003ci64: \\1\u003e/g\u0027 \\\r\n-e \u0027s/dimensions \u003d dense\u003c\\[\\([0-9]*\\), \\([0-9]*\\)\\]\u003e : tensor\u003c2xi64\u003e/dimensions \u003d array\u003ci64: \\1, \\2\u003e/g\u0027 \\\r\n-e \u0027s/rhs_dilation \u003d dense\u003c\\([0-9]*\\)\u003e : tensor\u003c2xi64\u003e/rhs_dilation \u003d array\u003ci64: \\1, \\1\u003e/g\u0027 \\\r\n-e \u0027s/window_strides \u003d dense\u003c\\([0-9]*\\)\u003e : tensor\u003c2xi64\u003e/window_strides \u003d array\u003ci64: \\1, \\1\u003e/g\u0027 \\\r\nmodel.mlir\r\n```\r\n\r\nci-extra: build_e2e_test_artifacts, test_benchmark_suites", "tree_diff": [ { "type": "modify", "old_id": "3c81790c8ddc5f3849ff6e1a2b8cfb6b918ead3e", "old_mode": 33188, "old_path": "build_tools/python/e2e_test_framework/models/tf_models.py", "new_id": "c616aa168e72b7fd02e6ab7d9c7398d7d6a9992b", "new_mode": 33188, "new_path": "build_tools/python/e2e_test_framework/models/tf_models.py" }, { "type": "modify", "old_id": "76515ce6f77575ca2070ab855c0a6d3de942c4fe", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/LegalizeCHLO.cpp", "new_id": "8e2739bfba4dca471757bd63a6f955d2eb1704db", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/LegalizeCHLO.cpp" }, { "type": "modify", "old_id": "f311a0e1ec22104b05f5e3527959fee783c4b5c8", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/LegalizeToLinalgUtils.h", "new_id": "6055b8ba25a1d1bcad54ef78a2f0d713c7d1f90f", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/LegalizeToLinalgUtils.h" }, { "type": "modify", "old_id": "fe6d413b1c39b396f9d2b3c959d4f4a96d633a96", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/Canonicalization.cpp", "new_id": "8158c211cbc3cf58ba7c47191259e0e85bd99e62", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/Canonicalization.cpp" }, { "type": "modify", "old_id": "58e39adfbfa9d1b5b2d09c3b5ac34b3ab73568a0", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/GatherToTorchIndexSelect.cpp", "new_id": "998c9d61f8173ec0c31928c862362a82f3da49ba", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/GatherToTorchIndexSelect.cpp" }, { "type": "modify", "old_id": "7fb89d144069548404fa9beb39a5b26fd085f865", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/StableHLOToStableHLO.cpp", "new_id": "0f3d82ecf993ccba829a9bbc73882fbfad4acae6", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/StableHLOToStableHLO.cpp" }, { "type": "modify", "old_id": "de696d6ef974d2b3104f0342cce16266f8a87a9a", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/UnfuseBatchNorm.cpp", "new_id": "3b0ed03da02c0504b04111962c7a07a93a3f128a", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/UnfuseBatchNorm.cpp" }, { "type": "modify", "old_id": "0f990b9e149725f2cebc90c095d4ba16e1d8a502", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/test/canonicalization.mlir", "new_id": "d5e8f540f28bbb031952951b2f6fc6a094ba3aca", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/test/canonicalization.mlir" }, { "type": "modify", "old_id": "79b944dd3a5826d1e2c893e6ff3a25abc939b1d1", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/test/gather_to_torch_index_select.mlir", "new_id": "c2530ae8b41702b05b02503cb722f9321fca538f", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/test/gather_to_torch_index_select.mlir" }, { "type": "modify", "old_id": "2ebad087ba81d2d657b4540ae55548c7f887c4e5", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/test/stablehlo_to_stablehlo.mlir", "new_id": "972067f1df238e6fe1fdd1fa70092fce6b8cf22b", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/Preprocessing/test/stablehlo_to_stablehlo.mlir" }, { "type": "modify", "old_id": "ff36eb135306d0cda5c751069b2fc115d877d86d", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/StableHLOToLinalg.cpp", "new_id": "e775a4475cd09579e0e3d427f17f045d7e7e4147", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/StableHLOToLinalg.cpp" }, { "type": "modify", "old_id": "08f77e685512503ed27b32a5f4acd5acc1b9fdd3", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/StableHLOToLinalgConvolution.cpp", "new_id": "5d68d62819acf89cd873d7f5a4e9f1ce5b3c3991", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/StableHLOToLinalgConvolution.cpp" }, { "type": "modify", "old_id": "0505d268cd8c3f340f1a98464683ff0aa04b6b62", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/StableHLOToLinalgExt.cpp", "new_id": "52c287df4e18b3892ce48f257cfee7e446d7b35a", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/StableHLOToLinalgExt.cpp" }, { "type": "modify", "old_id": "54744594a6fbb65c21f3cd949e3089f19876a740", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/StableHLOToLinalgReduce.cpp", "new_id": "632dbb5a2db5aee4e6dba5baceb9a00390575d54", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/StableHLOToLinalgReduce.cpp" }, { "type": "modify", "old_id": "9e6192c7ff3e488d8757285cc097d3e2ec01cffa", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/legalize_chlo_no_broadcast.mlir", "new_id": "3a6fbd12a353dd77e46b809fea1f3ccbedcb93f4", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/legalize_chlo_no_broadcast.mlir" }, { "type": "modify", "old_id": "95ea14a1ee3fd77c6354190ac0671c04729e8239", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/legalize_chlo_with_broadcast.mlir", "new_id": "6ee69caa49ec6b3847b916121d0394a4c6c55747", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/legalize_chlo_with_broadcast.mlir" }, { "type": "modify", "old_id": "069f67f20e90e3bb04a82fd8718a4890f0096a42", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/legalize_control_flow.mlir", "new_id": "3c158b1021f6b6558d847399b78e9c232e38d3d1", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/legalize_control_flow.mlir" }, { "type": "modify", "old_id": "bcf0c43cfd0668ae9f9056acb53609cda4282367", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg.mlir", "new_id": "52f7c58fe07d20963d25a5b4c172227b8a793d11", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg.mlir" }, { "type": "modify", "old_id": "2b2058637c1575d3b10be1dd9b19eb6b44a66ed3", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg_convolution.mlir", "new_id": "2e361bd0e32af5aa20e4f70663d6b47c6457cd9e", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg_convolution.mlir" }, { "type": "modify", "old_id": "ebd00192b4e80b20bf4abd57b82ea7e815211407", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg_ext.mlir", "new_id": "38792b3a9839826f896e7d786ea0febc38262650", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg_ext.mlir" }, { "type": "modify", "old_id": "c8003fc8a478c360956ddbf599a0f371972dd954", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg_gather.mlir", "new_id": "2156cd99304df04c92c211619efbaa64044b90f5", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg_gather.mlir" }, { "type": "modify", "old_id": "de2fd82065d7661f0ff73e8fb22215f6703ba562", "old_mode": 33188, "old_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg_reduce.mlir", "new_id": "f61474b1ec6f4485a57ba78958f3a1a88c50391f", "new_mode": 33188, "new_path": "compiler/plugins/input/StableHLO/stablehlo-iree/Conversion/test/stablehlo_to_linalg_reduce.mlir" }, { "type": "modify", "old_id": "577b47dc1895f0eda26187c1d61c8d0e2b697a7a", "old_mode": 33188, "old_path": "tests/e2e/regression/dynamic_reduce_min.mlir", "new_id": "382d268bd02f06dcba465a7b56b39e1f35e63225", "new_mode": 33188, "new_path": "tests/e2e/regression/dynamic_reduce_min.mlir" }, { "type": "modify", "old_id": "259eb01046c597c8625a17ddf98c17bea14bed81", "old_mode": 33188, "old_path": "tests/e2e/regression/lowering_config.mlir", "new_id": "766fbdf0b92c72f7c5a63e46c76b018caa5afd60", "new_mode": 33188, "new_path": "tests/e2e/regression/lowering_config.mlir" }, { "type": "modify", "old_id": "2e5bcf959bc477a21db7d0a37ab51c9989fbc90b", "old_mode": 33188, "old_path": "tests/e2e/stablehlo_models/mnist_train_test/mnist_train_test.py", "new_id": "f68c5d94fcaca73391730ee8e9b12fe01fdc5952", "new_mode": 33188, "new_path": "tests/e2e/stablehlo_models/mnist_train_test/mnist_train_test.py" }, { "type": "modify", "old_id": "980cbddf7e857d2c98d842163999cf17360ceb37", "old_mode": 33188, "old_path": "tests/e2e/stablehlo_models/unidirectional_lstm.mlir", "new_id": "420cddd09774892c74b1641581ad8e2a35e4837e", "new_mode": 33188, "new_path": "tests/e2e/stablehlo_models/unidirectional_lstm.mlir" }, { "type": "modify", "old_id": "0f92c522ffa86cab0fcff838297d929b493f638e", "old_mode": 33188, "old_path": "tests/e2e/stablehlo_ops/broadcast_in_dim.mlir", "new_id": "e78bd105d967ac4c78f4e3c3f4b1312f1303ad4e", "new_mode": 33188, "new_path": "tests/e2e/stablehlo_ops/broadcast_in_dim.mlir" }, { "type": "modify", "old_id": "b7fb713cd851f4a4c3a19e2be60766ddc8dcaf4c", "old_mode": 33188, "old_path": "tests/e2e/stablehlo_ops/convolution.mlir", "new_id": "57e223a61fb05119cd0628ae1beb4bfde6551e6e", "new_mode": 33188, "new_path": "tests/e2e/stablehlo_ops/convolution.mlir" }, { "type": "modify", "old_id": "af3cd0e30d1fa57dd9c9acf0b5ed88e7b553c15f", "old_mode": 33188, "old_path": "tests/e2e/stablehlo_ops/gather.mlir", "new_id": "6b9553651ad7ecaffd7897b18a61d319b8e3ab58", "new_mode": 33188, "new_path": "tests/e2e/stablehlo_ops/gather.mlir" }, { "type": "modify", "old_id": "fdad8971d69bb3c1b5c52dc57bceeff5dc683982", "old_mode": 33188, "old_path": "tests/e2e/stablehlo_ops/reduce.mlir", "new_id": "07e6aae51c682ee3f0f3153d3a1af5d9628a4e26", "new_mode": 33188, "new_path": "tests/e2e/stablehlo_ops/reduce.mlir" }, { "type": "modify", "old_id": "568d51c8c3718250775fcc780ed3743ecaa8d6b1", "old_mode": 33188, "old_path": "tests/e2e/stablehlo_ops/reduce_window.mlir", "new_id": "2d21ca788b561edd64f900f94c07adf41b117ea0", "new_mode": 33188, "new_path": "tests/e2e/stablehlo_ops/reduce_window.mlir" }, { "type": "modify", "old_id": "bbcc10fc3af385cc9ac6761665784600499f0fdf", "old_mode": 33188, "old_path": "tests/e2e/test_artifacts/generated_e2e_test_fetch_models.cmake", "new_id": "e26b098ea6b0f2a4f4125035d1d54f456429d76c", "new_mode": 33188, "new_path": "tests/e2e/test_artifacts/generated_e2e_test_fetch_models.cmake" }, { "type": "modify", "old_id": "6639a698acce2b449cda58ee9c21215a5d4668df", "old_mode": 33188, "old_path": "tests/e2e/test_artifacts/generated_e2e_test_iree_artifacts.cmake", "new_id": "57d2d49a6eb41207caea6ee6f810ec27909acbf8", "new_mode": 33188, "new_path": "tests/e2e/test_artifacts/generated_e2e_test_iree_artifacts.cmake" }, { "type": "modify", "old_id": "0cce02ee965704a0c32b472bd4eb4cf55bded96b", "old_mode": 33188, "old_path": "tests/e2e/vulkan_specific/conv.mlir", "new_id": "571a6ee6e674c8b37a6a2ac73481c7fb3552f9ee", "new_mode": 33188, "new_path": "tests/e2e/vulkan_specific/conv.mlir" }, { "type": "modify", "old_id": "b320a2f6ef38007723bd840a04141238e73497da", "old_mode": 33188, "old_path": "tests/microbenchmarks/stablehlo_conv.mlir", "new_id": "0318b6ca4c803534c0dbcdfe03c99dd4e4173361", "new_mode": 33188, "new_path": "tests/microbenchmarks/stablehlo_conv.mlir" }, { "type": "modify", "old_id": "f8dcebfa1ec166806974f6ae0dfb902d36b47238", "old_mode": 57344, "old_path": "third_party/stablehlo", "new_id": "0264c4d64c82ae74a54b85d274eec5084c2c0abf", "new_mode": 57344, "new_path": "third_party/stablehlo" } ] }