# Copyright 2017 The TensorFlow Authors. All Rights Reserved.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

# ==============================================================================

"""configure script to get build parameters from user."""

import argparse

import errno

import glob

import os

import platform

import re

import subprocess

import sys

# pylint: disable=g-import-not-at-top

try:

from shutil import which

except ImportError:

from distutils.spawn import find_executable as which

# pylint: enable=g-import-not-at-top

_DEFAULT_CUDA_VERSION = '11'

_DEFAULT_CUDNN_VERSION = '2'

_DEFAULT_TENSORRT_VERSION = '6'

_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '3.5,7.0'

_SUPPORTED_ANDROID_NDK_VERSIONS = [

19, 20, 21

]

_DEFAULT_PROMPT_ASK_ATTEMPTS = 10

_TF_BAZELRC_FILENAME = '.tf_configure.bazelrc'

_TF_WORKSPACE_ROOT = ''

_TF_BAZELRC = ''

_TF_CURRENT_BAZEL_VERSION = None

NCCL_LIB_PATHS = [

'lib64/', 'lib/powerpc64le-linux-gnu/', 'lib/x86_64-linux-gnu/', ''

]

# List of files to configure when building Bazel on Apple platforms.

APPLE_BAZEL_FILES = [

]

# List of files to move when building for iOS.

IOS_FILES = [

'tensorflow/lite/objc/TensorFlowLiteObjC.podspec',

'tensorflow/lite/swift/TensorFlowLiteSwift.podspec',

]

class UserInputError(Exception):

pass

def is_windows():

return platform.system() == 'Windows'

def is_linux():

return platform.system() == 'Linux'

def is_macos():

return platform.system() == 'Darwin'

def is_ppc64le():

return platform.machine() == 'ppc64le'

def is_s390x():

return platform.machine() == 's390x'

def is_cygwin():

return platform.system().startswith('CYGWIN_NT')

def get_input(question):

return answer

def symlink_force(target, link_name):

raise e

def write_to_bazelrc(line):

with open(_TF_BAZELRC, 'a') as f:

f.write(line + '\n')

def write_action_env_to_bazelrc(var_name, var):

write_to_bazelrc('build --action_env {}="{}"'.format(var_name, str(var)))

def run_shell(cmd, allow_non_zero=False, stderr=None):

return output.decode('UTF-8').strip()

def cygpath(path):

"""Convert path from posix to windows."""

return os.path.abspath(path).replace('\\', '/')

def get_python_path(environ_cp, python_bin_path):

return paths

def get_python_major_version(python_bin_path):

"""Get the python major version."""

return run_shell([python_bin_path, '-c', 'import sys; print(sys.version[0])'])

def setup_python(environ_cp):

f.write('export PYTHON_BIN_PATH="{}"'.format(python_bin_path))

def reset_tf_configure_bazelrc():

"""Reset file that contains customized config settings."""

open(_TF_BAZELRC, 'w').close()

def cleanup_makefile():

os.remove(os.path.join(root, f))

def get_var(environ_cp,

return var

def set_action_env_var(environ_cp,

environ_cp[var_name] = str(var)

def convert_version_to_int(version):

return int(version_str)

def retrieve_bazel_version():

return curr_version

def set_cc_opt_flags(environ_cp):

write_to_bazelrc('build:opt --host_copt=%s' % opt)

def set_tf_cuda_clang(environ_cp):

)

def set_tf_download_clang(environ_cp):

bazel_config_name='download_clang')

def get_from_env_or_user_or_default(environ_cp, var_name, ask_for_var,

return var

def prompt_loop_or_load_from_env(environ_cp,

return val

def set_clang_cuda_compiler_path(environ_cp):

return clang_cuda_compiler_path

def create_android_ndk_rule(environ_cp):

get_ndk_api_level(environ_cp, android_ndk_home_path))

def create_android_sdk_rule(environ_cp):

write_action_env_to_bazelrc('ANDROID_SDK_HOME', android_sdk_home_path)

def get_ndk_api_level(environ_cp, android_ndk_home_path):

return android_ndk_api_level

def set_gcc_host_compiler_path(environ_cp):

write_action_env_to_bazelrc('GCC_HOST_COMPILER_PATH', gcc_host_compiler_path)

def choose_compiler(environ_cp):

return var

def set_clang_compiler_path(environ_cp):

return clang_compiler_path

def retrieve_clang_version(clang_executable):

return curr_version

# Disable clang extension that rejects type definitions within offsetof.

# This was added in clang-16 by https://reviews.llvm.org/D133574.

# Can be removed once upb is updated, since a type definition is used within

# offset of in the current version of ubp. See

# https://github.com/protocolbuffers/upb/blob/9effcbcb27f0a665f9f345030188c0b291e32482/upb/upb.c#L183.

def disable_clang16_offsetof_extension(clang_version):

if int(clang_version.split('.')[0]) == 16:

write_to_bazelrc('build --copt=-Wno-gnu-offsetof-extensions')

def set_tf_cuda_paths(environ_cp):

environ_cp['TF_CUDA_PATHS'] = tf_cuda_paths

def set_tf_cuda_version(environ_cp):

environ_cp['TF_CUDA_VERSION'] = tf_cuda_version

def set_tf_cudnn_version(environ_cp):

environ_cp['TF_CUDNN_VERSION'] = tf_cudnn_version

def set_tf_tensorrt_version(environ_cp):

environ_cp['TF_TENSORRT_VERSION'] = tf_tensorrt_version

def set_tf_nccl_version(environ_cp):

environ_cp['TF_NCCL_VERSION'] = tf_nccl_version

def get_native_cuda_compute_capabilities(environ_cp):

return output

def set_tf_cuda_compute_capabilities(environ_cp):