diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-24 17:20:07 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-24 17:24:32 -0700 |
| commit | 73083d29afe770870742a9d19555686886e76f6d (patch) | |
| tree | c18fa3b5665b2dd2d069104387531bfc2a1011c4 /tensorflow/contrib/tpu | |
| parent | cc5555d3d3daa64f462cc7f8d31fe915073429f4 (diff) | |
Upgrade cloud tpu profiler to 1.11.
PiperOrigin-RevId: 214359786
Diffstat (limited to 'tensorflow/contrib/tpu')
| -rw-r--r-- | tensorflow/contrib/tpu/profiler/pip_package/cloud_tpu_profiler/main.py | 7 | ||||
| -rw-r--r-- | tensorflow/contrib/tpu/profiler/pip_package/setup.py | 2 | ||||
| -rw-r--r-- | 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_ |