Update orttraining/orttraining/core/optimizer/compute_optimizer/paddi…

…ng_elimination.cc
microsoft · Sep 5, 2024 · 17e8d5c · 17e8d5c
1 parent 019b4af
commit 17e8d5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/orttraining/orttraining/core/optimizer/compute_optimizer/padding_elimination.cc b/orttraining/orttraining/core/optimizer/compute_optimizer/padding_elimination.cc
@@ -285,7 +285,7 @@ void IterateSubgraphFromNode(Graph& graph,
       PushAllOutputNode(graph, to_visit, cur, visited);
     } else if (graph_utils::IsSupportedOptypeVersionAndDomain(*cur, "Cast", {9, 13}) ||
                graph_utils::IsSupportedOptypeVersionAndDomain(*cur, "FastGelu", {1}, kMSDomain) ||
-               graph_utils::IsSupportedOptypeVersionAndDomain(*cur, "Gelu", {1, 20}) ||
+               graph_utils::IsSupportedOptypeVersionAndDomain(*cur, "Gelu", {20}) ||
                graph_utils::IsSupportedOptypeVersionAndDomain(*cur, "Gelu", {1}, kMSDomain) ||
                graph_utils::IsSupportedOptypeVersionAndDomain(*cur, "QuickGelu", {1}, kMSDomain) ||
                graph_utils::IsSupportedOptypeVersionAndDomain(*cur, "Sqrt", {6, 13})) {