Merge commit for internal changes

This commit is contained in:
Shanqing Cai 2017-05-17 11:53:56 -04:00
commit 2309260983
5 changed files with 18 additions and 5 deletions
tensorflow
contrib/cmake
python
framework
kernel_tests
tensorboard/components/tf_distribution_dashboard
third_party/gpus

View File

@ -181,6 +181,7 @@ if (tensorflow_BUILD_PYTHON_TESTS)
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/cast_op_test.py"
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/string_to_number_op_test.py"
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/clip_ops_test.py"
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/tensor_array_ops_test.py" # Needs portpicker.
# misc
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/variable_scope_test.py"
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/reshape_op_test.py"

View File

@ -27,9 +27,14 @@ import tempfile
import threading
import numpy as np
import portpicker
import six
try:
import portpicker # pylint: disable=g-import-not-at-top
except ImportError as _portpicker_import_error:
portpicker = None
# pylint: disable=g-import-not-at-top
from google.protobuf import descriptor_pool
from google.protobuf import text_format
@ -811,7 +816,12 @@ def create_local_cluster(num_workers, num_ps, protocol="grpc"):
A tuple `(worker_servers, ps_servers)`. `worker_servers` is a list
of `num_workers` objects of type `tf.train.Server` (all running locally);
and `ps_servers` is a list of `num_ps` objects of similar type.
Raises:
ImportError: if portpicker module was not found at load time
"""
if not portpicker:
raise _portpicker_import_error
worker_ports = [portpicker.pick_unused_port() for _ in range(num_workers)]
ps_ports = [portpicker.pick_unused_port() for _ in range(num_ps)]
cluster_dict = {

View File

@ -1905,6 +1905,7 @@ cuda_py_test(
"//tensorflow/python:tensor_array_ops",
"//tensorflow/python:variables",
],
flaky = 1, # create_local_cluster sometimes times out.
)
cuda_py_test(

View File

@ -59,5 +59,6 @@ tensorboard_ts_library(
srcs = [],
deps_mgmt = "off",
runtime = "nodejs",
deps = ["//tensorflow/tensorboard/components:common_deps"],
# Don't list the common_deps here; a src-less ts_library re-exports them,
# and they are outside the package where the Fileset will be rooted.
)

View File

@ -787,7 +787,7 @@ def _symlink_genrule_for_dir(repository_ctx, src_dir, dest_dir, genrule_name,
if dest_files[i] != "":
# If we have only one file to link we do not want to use the dest_dir, as
# $(@D) will include the full path to the file.
dest = ' $(@D)/' + dest_dir + dest_files[i] if len(dest_files) != 1 else ' $(@D)' + dest_files[i]
dest = ' $(@D)/' + dest_dir + dest_files[i] if len(dest_files) != 1 else ' $(@D)/' + dest_files[i]
command.append('ln -s ' + src_files[i] + dest)
outs.append(' "' + dest_dir + dest_files[i] + '",')
genrule = _genrule(src_dir, genrule_name, " && ".join(command),
@ -875,8 +875,8 @@ def _create_cuda_repository(repository_ctx):
included_files = _read_dir(repository_ctx, cuda_include_path).replace(
cuda_include_path, '').splitlines()
if '/cudnn.h' not in included_files:
genrules.append(_symlink_genrule_for_dir(repository_ctx, None, "",
"cudnn-include", [cudnn_header_dir + "/cudnn.h"], ["include/cudnn.h"]))
genrules.append(_symlink_genrule_for_dir(repository_ctx, None, "include/",
"cudnn-include", [cudnn_header_dir + "/cudnn.h"], ["cudnn.h"]))
else:
genrules.append(
'filegroup(\n' +