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
// xla.compile() Python code into XlaLaunch nodes.
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 26,
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 36,
EncapsulateXlaComputationsPass);
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 25,
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 35,
IntroduceFloatingPointJitterPass);
// from

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

@ -17,7 +17,7 @@ limitations under the License.
namespace tensorflow {
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0,
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10,
MlirImportPass);
} // 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 {
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 0,
REGISTER_OPTIMIZATION(OptimizationPassRegistry::PRE_PLACEMENT, 10,
MlirRoundtripPass);
} // 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
// 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);
} // namespace tensorflow

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

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

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

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

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

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

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

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