From b627bc9019f03a6faba2d5d5fa1fe8ebda066cbb Mon Sep 17 00:00:00 2001 From: Cheng Tang Date: Fri, 3 Jul 2020 17:29:03 -0700 Subject: [PATCH] temp disable gpu fusing transformers --- .../orttraining/core/optimizer/graph_transformer_utils.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/orttraining/orttraining/core/optimizer/graph_transformer_utils.cc b/orttraining/orttraining/core/optimizer/graph_transformer_utils.cc index a9ac30528a..8802f3e70d 100644 --- a/orttraining/orttraining/core/optimizer/graph_transformer_utils.cc +++ b/orttraining/orttraining/core/optimizer/graph_transformer_utils.cc @@ -71,7 +71,7 @@ std::vector> GeneratePreTrainingTransformers(T transformers.emplace_back(onnxruntime::make_unique(compatible_eps)); transformers.emplace_back(onnxruntime::make_unique(compatible_eps)); - transformers.emplace_back(onnxruntime::make_unique(compatible_eps)); + //transformers.emplace_back(onnxruntime::make_unique(compatible_eps)); if (enable_gelu_approximation) { transformers.emplace_back(onnxruntime::make_unique(compatible_eps)); @@ -144,7 +144,7 @@ std::vector> GenerateTransformers(TransformerL transformers.emplace_back(onnxruntime::make_unique(l1_execution_providers)); transformers.emplace_back(onnxruntime::make_unique(free_dimension_overrides)); transformers.emplace_back(onnxruntime::make_unique(l1_execution_providers)); - transformers.emplace_back(onnxruntime::make_unique(l1_execution_providers)); + //transformers.emplace_back(onnxruntime::make_unique(l1_execution_providers)); rule_transformer = optimizer_utils::GenerateRuleBasedGraphTransformer(level, transformers_and_rules_to_enable, l1_execution_providers); } break;