Skip to content

Commit fdfda25

Browse files
committed
Give the StableHLO Aggressive Simp. patterns names (openxla#2780)
This improves stack traces as well as general code readability.
1 parent 2199d09 commit fdfda25

File tree

3 files changed

+239
-129
lines changed

3 files changed

+239
-129
lines changed

.github/workflows/buildAndTestBazel.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,10 +56,10 @@ jobs:
5656
echo "version=$(cat ${{ github.workspace }}/build_tools/llvm_version.txt)" >> $GITHUB_OUTPUT
5757
5858
- name: Setup Bazelisk
59-
uses: bazelbuild/setup-bazelisk@v2
59+
uses: bazelbuild/setup-bazelisk@v3
6060

6161
- name: Mount Bazel Cache
62-
uses: actions/cache@v3
62+
uses: actions/cache@v4
6363
with:
6464
path: "~/.cache/bazel"
6565
key: ${{ runner.os }}-stablehlo_bazelbuild-${{ hashFiles('.bazelversion', '.bazelrc', 'WORKSPACE', 'WORKSPACE.bazel', 'MODULE.bazel') }}

stablehlo/transforms/optimization/StablehloAggressiveSimplification.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1539,8 +1539,8 @@ void populateStablehloCanonicalizationPatterns(MLIRContext *context,
15391539

15401540
void populateStablehloHloImportCanonicalizationPatterns(
15411541
MLIRContext *context, RewritePatternSet *patterns) {
1542-
patterns->add<ReshapeIsNoop, TupleIsRepacking, TupleIsUnpacked,
1543-
WhileOpImplicitCapture>(context);
1542+
patterns->add<ReshapeOp_RemoveNoop, GetTupleElementOp_UnpackTuple,
1543+
TupleIsRepacking, WhileOpImplicitCapture>(context);
15441544
}
15451545

15461546
std::unique_ptr<Pass> createStablehloAggressiveSimplificationPass(

0 commit comments

Comments
 (0)