Merge commit for internal changes
This commit is contained in:
commit
2309260983
tensorflow
contrib/cmake
python
tensorboard/components/tf_distribution_dashboard
third_party/gpus
@ -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"
|
||||
|
@ -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 = {
|
||||
|
@ -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(
|
||||
|
@ -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.
|
||||
)
|
||||
|
6
third_party/gpus/cuda_configure.bzl
vendored
6
third_party/gpus/cuda_configure.bzl
vendored
@ -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' +
|
||||
|
Loading…
Reference in New Issue
Block a user