Skip to content

Commit

Permalink
fix the warning
Browse files Browse the repository at this point in the history
  • Loading branch information
ruiren committed Feb 7, 2024
1 parent 7870789 commit 5a7e0c0
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions onnxruntime/core/optimizer/gather_slice_fusion.cc
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,7 @@ Status GatherSliceToSplitFusion::ApplyImpl(Graph& graph, bool& modified, int gra
InlinedVector<NodeArg*> split_outputs(3);

InlinedVector<std::reference_wrapper<Node>> nodes_to_fuse;
int64_t gather_node_count = 0, slice_node_count = 0;
size_t gather_node_count = 0, slice_node_count = 0;

// find the nodes to be merged
for (auto consumer : consumers) {
Expand Down Expand Up @@ -260,8 +260,7 @@ Status GatherSliceToSplitFusion::ApplyImpl(Graph& graph, bool& modified, int gra
Node& gather_node = *graph.GetNode(consumer->Index());
nodes_to_fuse.push_back(gather_node);
NodeArg* gather_output_args = gather_node.MutableOutputDefs()[0];
gather_node_count++;
split_outputs[gather_node_count] = gather_output_args;
split_outputs[++gather_node_count] = gather_output_args;
}

// check the Slice Ops
Expand Down

0 comments on commit 5a7e0c0

Please sign in to comment.