experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix typos from the latest version

Browse Source
This commit is contained in:
Austin Anderson 2020-04-30 17:45:36 -07:00
parent 777482a626
commit 642dc30aca
3 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/tensorflow.bzl
View File

@ -2608,7 +2608,7 @@ def tf_py_build_info_genrule(name, out):
+ " is_rocm_build=" + if_rocm("True", "False")
+ " is_cuda_build=" + if_cuda("True", "False")
# TODO(angerson) Can we reliably load CUDA compute capabilities here?
if_windows(dict_to_kv({
+ if_windows(dict_to_kv({
"msvcp_dll_names": "msvcp140.dll,msvcp140_1.dll"
}), "") + if_windows_cuda(dict_to_kv({
"nvcuda_dll_name": "nvcuda.dll",

2
tensorflow/tools/build_info/gen_build_info.py
View File

@ -84,7 +84,7 @@ def write_build_info(filename, key_value_list):
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
\"\"\"Auto-generated module providing information about the build.\"\"\""
\"\"\"Auto-generated module providing information about the build.\"\"\"
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

14
tensorflow/tools/pip_package/setup.py
View File

@ -85,16 +85,18 @@ REQUIRED_PACKAGES = [
'scipy == 1.2.2;python_version<"3"',
]
# Generate a footer describing the CUDA technology this release was built
# against.
GPU_DESCRIPTION = ''
if build_info.build_info['is_cuda_build']:
gpu_header = (f'\nTensorFlow {_VERSION} for NVIDIA GPUs was built with these '
'platform and library versions:\n\n - ')
cbi = build_info.cuda_build_info
trt_ver = cbi['tensorrt_version']
nccl_ver = cbi['nccl_version']
bi = build_info.build_info
trt_ver = bi['tensorrt_version']
nccl_ver = bi['nccl_version']
GPU_DESCRIPTION = gpu_header + '\n - '.join([
'NVIDIA CUDA ' + cbi['cuda_version'],
'NVIDIA cuDNN ' + cbi['cudnn_version'],
'NVIDIA CUDA ' + bi['cuda_version'],
'NVIDIA cuDNN ' + bi['cudnn_version'],
'NVIDIA NCCL ' + 'not enabled' if not nccl_ver else nccl_ver,
'NVIDIA TensorRT ' + 'not enabled' if not trt_ver else trt_ver,
])
@ -303,7 +305,7 @@ setup(
zip_safe=False,
# Accessible with importlib.metadata.metadata('tf-pkg-name').items()
platforms=[
f'{key}:{value}' for key, value in build_info.cuda_build_info.items()
f'{key}:{value}' for key, value in build_info.build_info.items()
],
distclass=BinaryDistribution,
cmdclass={