| commit | c909f5e74a38c5ffbe93a6b78cb435205d27b1a7 | [log] [tgz] |
|---|---|---|
| author | A. Unique TensorFlower <gardener@tensorflow.org> | Mon May 04 03:15:45 2020 -0700 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Mon May 04 03:18:54 2020 -0700 |
| tree | f6ca2dc20ac322cde77ffa9ca3e4288bf99efbd9 | |
| parent | 0bd921a425f01ea5e696d225f8b2e2b9569a2a83 [diff] |
Integrate LLVM at https://github.com/llvm/llvm-project/commit/6fe20a44fd3f PiperOrigin-RevId: 309710480 Change-Id: Iffb04c768392b9803d2f3c2c6a3ad23f9489cf86
diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc b/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc index 8b61280..a01769b 100644 --- a/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc +++ b/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
@@ -30,7 +30,7 @@ void EnableLogging(PassManager *pm) { // Print the whole module after each pass, which requires disabling // multi-threading as well. - pm->disableMultithreading(); + pm->getContext()->disableMultithreading(); pm->enableIRPrinting(std::make_unique<tensorflow::BridgeLoggerConfig>( /*print_module_scope=*/true)); pm->enableTiming(std::make_unique<tensorflow::BridgeTimingConfig>());
diff --git a/tensorflow/compiler/mlir/tensorflow/utils/compile_mlir_util.cc b/tensorflow/compiler/mlir/tensorflow/utils/compile_mlir_util.cc index b891682..7849213 100644 --- a/tensorflow/compiler/mlir/tensorflow/utils/compile_mlir_util.cc +++ b/tensorflow/compiler/mlir/tensorflow/utils/compile_mlir_util.cc
@@ -299,7 +299,7 @@ if (VLOG_IS_ON(1)) { // Print the whole module after each pass which requires disabling // multi-threading as well. - tf2xla.disableMultithreading(); + module_op.getContext()->disableMultithreading(); tf2xla.enableIRPrinting(std::make_unique<tensorflow::BridgeLoggerConfig>( /*print_module_scope=*/true)); }