Skip to content

Commit

Permalink
Enable test and make it pass.
Browse files Browse the repository at this point in the history
According to the test name, we want to ensure that DynamicUpdateSlice is not
fused into the reduce fusion. There is also a test expectation that there are
only 2 fusions, but my guess is that this is from before where we actually did
fuse DynamicUpdateSlice into the reduce fusion.

PiperOrigin-RevId: 726025657
  • Loading branch information
akuegel authored and Google-ML-Automation committed Feb 12, 2025
1 parent 9261575 commit 4c05b31
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions xla/service/gpu/transforms/priority_fusion_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -465,7 +465,6 @@ CHECK-NOT: fusion(
}

TEST_F(PriorityFusionTest, DoNotFuseDynamicUpdateSliceIntoReduce) {
GTEST_SKIP() << "b/294198633";
absl::string_view kHlo = R"(
HloModule test_module
Expand Down Expand Up @@ -533,11 +532,11 @@ ENTRY main {
})";

RunAndFilecheckHloRewrite(kHlo, std::move(priority_fusion_), R"(
CHECK: ROOT {{.*}} dynamic-update-slice(
CHECK: %[[REDUCE:.*]] = {{.*}} reduce(
CHECK: ROOT {{.*}} log(%[[REDUCE]])
CHECK: ROOT {{.*}} dynamic-update-slice(
CHECK: ENTRY
CHECK-COUNT-2: fusion(
CHECK-COUNT-3: fusion(
)");
}

Expand Down

0 comments on commit 4c05b31

Please sign in to comment.