#sdy unskip test and change to correct result now that we resolve rea…

…l conflicts. PiperOrigin-RevId: 696945866
openxla · Nov 15, 2024 · 2d07e59 · 2d07e59
1 parent a7aff7f
commit 2d07e59
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/xla/service/spmd/shardy/shardy_xla_pass_test.cc b/xla/service/spmd/shardy/shardy_xla_pass_test.cc
@@ -246,7 +246,6 @@ TEST_F(ShardyXLATest, DotMergeOperands1) {
 }
 
 TEST_F(ShardyXLATest, DotMergeOperands2) {
-  GTEST_SKIP() << "Temporarily skip failing test until Shardy fix is in.";
   const char* const hloString = R"(
     HloModule module
     ENTRY %conv {
@@ -267,7 +266,7 @@ TEST_F(ShardyXLATest, DotMergeOperands2) {
               op::Sharding("{devices=[2,2,2]<=[8]}"));
 
   EXPECT_THAT(module->entry_computation()->root_instruction(),
-              op::Sharding("{devices=[2,1,1,4]<=[8] last_tile_dim_replicate}"));
+              op::Sharding("{devices=[2,2,1,2]<=[8] last_tile_dim_replicate}"));
 }
 
 TEST_F(ShardyXLATest, DotMergeOperands3) {