update test and format

microsoft · Feb 7, 2024 · 7870789 · 7870789
1 parent abf2994
commit 7870789
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/onnxruntime/core/optimizer/gather_slice_fusion.cc b/onnxruntime/core/optimizer/gather_slice_fusion.cc
@@ -292,9 +292,8 @@ Status GatherSliceToSplitFusion::ApplyImpl(Graph& graph, bool& modified, int gra
     for (int64_t i = 0; i < rank; i++) {
       if (i == split_axis)
         split_output_type.mutable_tensor_type()->mutable_shape()->add_dim()->set_dim_value(1LL);
-      else {
+      else
         *(split_output_type.mutable_tensor_type()->mutable_shape()->add_dim()) = shape->dim(static_cast<int>(i));
-      }
     }
 
     InlinedVector<NodeArg*> split_output_types;

diff --git a/onnxruntime/test/optimizer/graph_transform_test.cc b/onnxruntime/test/optimizer/graph_transform_test.cc
@@ -7718,7 +7718,7 @@ TEST_F(GraphTransformationTests, GatherSliceToSplitFusion) {
     std::unique_ptr<GraphTransformer> transformer = std::make_unique<GatherSliceToSplitFusion>();
     ASSERT_STATUS_OK(TestGraphTransformer(build_test_case, 14, *logger_, std::move(transformer),
                                           TransformerLevel::Level1, 1, pre_graph_checker, post_graph_checker));
-  };
+  }
 }
 
 TEST_F(GraphTransformationTests, GatherSliceToSplitFusion_Invalid) {
@@ -7763,7 +7763,7 @@ TEST_F(GraphTransformationTests, GatherSliceToSplitFusion_Invalid) {
     };
 
     auto pre_graph_checker = [&](Graph& graph) {
-      TEST_RETURN_IF_NOT(CountOpsInGraph(graph)["Gather"] == 2);
+      TEST_RETURN_IF_NOT(CountOpsInGraph(graph)["Gather"] == 1);
       TEST_RETURN_IF_NOT(CountOpsInGraph(graph)["Slice"] == 1);
       return Status::OK();
     };
@@ -7778,7 +7778,7 @@ TEST_F(GraphTransformationTests, GatherSliceToSplitFusion_Invalid) {
     std::unique_ptr<GraphTransformer> transformer = std::make_unique<GatherSliceToSplitFusion>();
     ASSERT_STATUS_OK(TestGraphTransformer(build_test_case, 14, *logger_, std::move(transformer),
                                           TransformerLevel::Level1, 1, pre_graph_checker, post_graph_checker));
-  };
+  }
 }
 
 }  // namespace test