From 73083d29afe770870742a9d19555686886e76f6d Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 24 Sep 2018 17:20:07 -0700 Subject: Upgrade cloud tpu profiler to 1.11. PiperOrigin-RevId: 214359786 --- .../contrib/tpu/profiler/pip_package/cloud_tpu_profiler/main.py | 7 ++++--- tensorflow/contrib/tpu/profiler/pip_package/setup.py | 2 +- tensorflow/contrib/tpu/profiler/version.h | 2 +- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/main.py b/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/main.py index 438f442848..63641e00c5 100644 --- a/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/main.py +++ b/tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/main.py @@ -116,12 +116,13 @@ def main(unused_argv=None): elif tpu_cluster_resolver is not None: workers_list = get_workers_list(tpu_cluster_resolver) - if not FLAGS.logdir: + if not FLAGS.logdir and not FLAGS.monitoring_level: sys.exit('logdir must be provided.') executable_path = os.path.join(os.path.dirname(__file__), EXECUTABLE) - logdir = os.path.expandvars(os.path.expanduser(FLAGS.logdir)) cmd = [executable_path] - cmd.append('--logdir=' + logdir) + if FLAGS.logdir is not None: + logdir = os.path.expandvars(os.path.expanduser(FLAGS.logdir)) + cmd.append('--logdir=' + logdir) cmd.append('--service_addr=' + service_addr) cmd.append('--workers_list=' + workers_list) cmd.append('--duration_ms=' + str(FLAGS.duration_ms)) diff --git a/tensorflow/contrib/tpu/profiler/pip_package/setup.py b/tensorflow/contrib/tpu/profiler/pip_package/setup.py index d4ccb0f246..2415c46718 100644 --- a/tensorflow/contrib/tpu/profiler/pip_package/setup.py +++ b/tensorflow/contrib/tpu/profiler/pip_package/setup.py @@ -20,7 +20,7 @@ from __future__ import print_function from setuptools import setup -_VERSION = '1.10.0' +_VERSION = '1.11.0' CONSOLE_SCRIPTS = [ 'capture_tpu_profile=cloud_tpu_profiler.main:run_main', diff --git a/tensorflow/contrib/tpu/profiler/version.h b/tensorflow/contrib/tpu/profiler/version.h index aee094177b..90d34b5ef1 100644 --- a/tensorflow/contrib/tpu/profiler/version.h +++ b/tensorflow/contrib/tpu/profiler/version.h @@ -16,6 +16,6 @@ limitations under the License. #ifndef TENSORFLOW_CONTRIB_TPU_PROFILER_VERSION_H_ #define TENSORFLOW_CONTRIB_TPU_PROFILER_VERSION_H_ -#define TPU_PROFILER_VERSION "1.10.0" +#define TPU_PROFILER_VERSION "1.11.0" #endif // TENSORFLOW_CONTRIB_TPU_PROFILER_VERSION_H_ -- cgit v1.2.3