experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Move MLIR bridge pass to before all other passes.

Browse Source 
New order:
MlirBridgePass: Phase 0 before all passes
Passes that were at Phase 0 originally are moved to Phase 10
Passes that were at Phase 1 originally are moved to Phase 20
Passes that were at Phase 20+ originally are moved to Phase 30+
PiperOrigin-RevId: 282394988
Change-Id: Ief93072c52fcc073ceb0998271e1e4d5ad2d1f74
This commit is contained in:
Yanan Cao 2019-11-25 11:12:43 -08:00 committed by TensorFlower Gardener
parent b5d2f3677f
commit 1e71cf27dc
8 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/jit/jit_compilation_pass_registration.cc
View File

@ -31,10 +31,10 @@ namespace tensorflow {
// EncapsulateXlaComputationsPass rewrites computations generated by the // EncapsulateXlaComputationsPass rewrites computations generated by the
// xla.compile() Python code into XlaLaunch nodes. // xla.compile() Python code into XlaLaunch nodes.
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 26, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 36,
EncapsulateXlaComputationsPass); EncapsulateXlaComputationsPass);
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 25, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 35,
IntroduceFloatingPointJitterPass); IntroduceFloatingPointJitterPass);
// from // from

2
tensorflow/compiler/mlir/tensorflow/translate/mlir_import_pass_registration.cc
View File

@ -17,7 +17,7 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10,
MlirImportPass); MlirImportPass);
} // namespace tensorflow } // namespace tensorflow

2
tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_pass_registration.cc
View File

@ -17,7 +17,7 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10,
MlirRoundtripPass); MlirRoundtripPass);
} // namespace tensorflow } // namespace tensorflow

2
tensorflow/compiler/tf2xla/functionalize_control_flow_pass_registration.cc
View File

@ -19,7 +19,7 @@ namespace tensorflow {
// This pass is required for some AOT backends and all JIT backends, so this // This pass is required for some AOT backends and all JIT backends, so this
// file exists as a separate lib and will be linked to both AOT and JIT. // file exists as a separate lib and will be linked to both AOT and JIT.
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 27, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 37,
FunctionalizeControlFlowForXlaPass); FunctionalizeControlFlowForXlaPass);
} // namespace tensorflow } // namespace tensorflow

2
tensorflow/core/common_runtime/accumulate_n_optimizer.cc
View File

@ -284,7 +284,7 @@ class AccumulateNV2RemovePass : public GraphOptimizationPass {
return Status::OK(); return Status::OK();
} }
}; };
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10,
AccumulateNV2RemovePass); AccumulateNV2RemovePass);
} // namespace } // namespace

2
tensorflow/core/common_runtime/isolate_placer_inspection_required_ops_pass.cc
View File

@ -48,7 +48,7 @@ Status IsolatePlacerInspectionRequiredOpsPass::Run(
return status; return status;
} }
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 25, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 35,
IsolatePlacerInspectionRequiredOpsPass); IsolatePlacerInspectionRequiredOpsPass);
} // namespace tensorflow } // namespace tensorflow

2
tensorflow/core/common_runtime/lower_functional_ops.cc
View File

@ -179,7 +179,7 @@ Status LowerFunctionalOpsPass::Run(
return Status::OK(); return Status::OK();
} }
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10,
LowerFunctionalOpsPass); LowerFunctionalOpsPass);
} // namespace tensorflow } // namespace tensorflow

2
tensorflow/core/common_runtime/parallel_concat_optimizer.cc
View File

@ -117,7 +117,7 @@ class ParallelConcatRemovePass : public GraphOptimizationPass {
return Status::OK(); return Status::OK();
} }
}; };
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0, REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10,
ParallelConcatRemovePass); ParallelConcatRemovePass);
} // namespace } // namespace