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

Enable logging by default about running or not the MLIR bridge

Browse Source 
PiperOrigin-RevId: 309344310
Change-Id: I610ab30c0483156e7bcc16185998005dd7da264f
This commit is contained in:
Mehdi Amini 2020-04-30 19:12:05 -07:00 committed by TensorFlower Gardener
parent 7c26347d2e
commit da4a324454
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/compiler/tf2xla/mlir_bridge_pass.cc
View File

@ -31,11 +31,11 @@ namespace tensorflow {
Status MlirBridgePass::Run(const ConfigProto& config_proto,
mlir::ModuleOp module) {
if (!config_proto.experimental().enable_mlir_bridge()) {
VLOG(1) << "Skipping MLIR Bridge Pass, session flag not enabled";
VLOG(0) << "Skipping MLIR TPU Bridge, session flag not enabled";
return Status::OK();
}
VLOG(1) << "Running MLIR Bridge Pass";
VLOG(0) << "Running MLIR TPU Bridge";
TF_RETURN_IF_ERROR(
mlir::TFTPU::TPUBridge(module, /*enable_logging=*/VLOG_IS_ON(1)));
@ -47,11 +47,11 @@ Status MlirBridgeV1CompatPass::Run(const GraphOptimizationPassOptions& options,
if (options.is_function_graph) return Status::OK();
if (!options.session_options->config.experimental().enable_mlir_bridge()) {
VLOG(1) << "Skipping MLIR Bridge V1 Compat Pass, session flag not enabled";
VLOG(0) << "Skipping MLIR TPU Bridge V1 Compat, session flag not enabled";
return Status::OK();
}
VLOG(1) << "Running MLIR Bridge V1 Compat Pass";
VLOG(0) << "Running MLIR TPU Bridge V1 Compat";
TF_RETURN_IF_ERROR(
mlir::TFTPU::TPUBridgeV1Compat(module, /*enable_logging=*/VLOG_IS_ON(1)));