Merge commit for internal changes
This commit is contained in:
commit
2309260983
@ -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/cast_op_test.py"
|
||||||
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/string_to_number_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/clip_ops_test.py"
|
||||||
|
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/tensor_array_ops_test.py" # Needs portpicker.
|
||||||
# misc
|
# misc
|
||||||
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/variable_scope_test.py"
|
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/variable_scope_test.py"
|
||||||
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/reshape_op_test.py"
|
"${tensorflow_source_dir}/tensorflow/python/kernel_tests/reshape_op_test.py"
|
||||||
|
@ -27,9 +27,14 @@ import tempfile
|
|||||||
import threading
|
import threading
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import portpicker
|
|
||||||
import six
|
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 descriptor_pool
|
||||||
from google.protobuf import text_format
|
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
|
A tuple `(worker_servers, ps_servers)`. `worker_servers` is a list
|
||||||
of `num_workers` objects of type `tf.train.Server` (all running locally);
|
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.
|
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)]
|
worker_ports = [portpicker.pick_unused_port() for _ in range(num_workers)]
|
||||||
ps_ports = [portpicker.pick_unused_port() for _ in range(num_ps)]
|
ps_ports = [portpicker.pick_unused_port() for _ in range(num_ps)]
|
||||||
cluster_dict = {
|
cluster_dict = {
|
||||||
|
@ -1905,6 +1905,7 @@ cuda_py_test(
|
|||||||
"//tensorflow/python:tensor_array_ops",
|
"//tensorflow/python:tensor_array_ops",
|
||||||
"//tensorflow/python:variables",
|
"//tensorflow/python:variables",
|
||||||
],
|
],
|
||||||
|
flaky = 1, # create_local_cluster sometimes times out.
|
||||||
)
|
)
|
||||||
|
|
||||||
cuda_py_test(
|
cuda_py_test(
|
||||||
|
@ -59,5 +59,6 @@ tensorboard_ts_library(
|
|||||||
srcs = [],
|
srcs = [],
|
||||||
deps_mgmt = "off",
|
deps_mgmt = "off",
|
||||||
runtime = "nodejs",
|
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 dest_files[i] != "":
|
||||||
# If we have only one file to link we do not want to use the dest_dir, as
|
# 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.
|
# $(@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)
|
command.append('ln -s ' + src_files[i] + dest)
|
||||||
outs.append(' "' + dest_dir + dest_files[i] + '",')
|
outs.append(' "' + dest_dir + dest_files[i] + '",')
|
||||||
genrule = _genrule(src_dir, genrule_name, " && ".join(command),
|
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(
|
included_files = _read_dir(repository_ctx, cuda_include_path).replace(
|
||||||
cuda_include_path, '').splitlines()
|
cuda_include_path, '').splitlines()
|
||||||
if '/cudnn.h' not in included_files:
|
if '/cudnn.h' not in included_files:
|
||||||
genrules.append(_symlink_genrule_for_dir(repository_ctx, None, "",
|
genrules.append(_symlink_genrule_for_dir(repository_ctx, None, "include/",
|
||||||
"cudnn-include", [cudnn_header_dir + "/cudnn.h"], ["include/cudnn.h"]))
|
"cudnn-include", [cudnn_header_dir + "/cudnn.h"], ["cudnn.h"]))
|
||||||
else:
|
else:
|
||||||
genrules.append(
|
genrules.append(
|
||||||
'filegroup(\n' +
|
'filegroup(\n' +
|
||||||
|
Loading…
x
Reference in New Issue
Block a user