2021-08-10 16:17:30 -07:00
|
|
|
############################################################################
|
|
|
|
|
# All default build options below.
|
|
|
|
|
|
2023-02-14 21:24:27 +00:00
|
|
|
# Required by OpenXLA
|
|
|
|
|
# https://github.com/openxla/xla/issues/1323
|
|
|
|
|
build --nocheck_visibility
|
|
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
# Sets the default Apple platform to macOS.
|
|
|
|
|
build --apple_platform_type=macos
|
2022-06-01 20:27:42 -04:00
|
|
|
build --macos_minimum_os=10.14
|
2021-08-10 16:17:30 -07:00
|
|
|
|
|
|
|
|
# Make Bazel print out all options from rc files.
|
|
|
|
|
build --announce_rc
|
|
|
|
|
|
|
|
|
|
build --define open_source_build=true
|
|
|
|
|
|
|
|
|
|
build --spawn_strategy=standalone
|
2021-08-18 13:56:08 -07:00
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
build --enable_platform_specific_config
|
|
|
|
|
|
2022-03-31 23:07:01 -07:00
|
|
|
build --experimental_cc_shared_library
|
|
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
# Disable enabled-by-default TensorFlow features that we don't care about.
|
|
|
|
|
build --define=no_aws_support=true
|
|
|
|
|
build --define=no_gcp_support=true
|
|
|
|
|
build --define=no_hdfs_support=true
|
|
|
|
|
build --define=no_kafka_support=true
|
|
|
|
|
build --define=no_ignite_support=true
|
|
|
|
|
|
|
|
|
|
build --define=grpc_no_ares=true
|
|
|
|
|
|
2022-10-13 13:50:09 +00:00
|
|
|
build --define=tsl_link_protobuf=true
|
|
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
build -c opt
|
|
|
|
|
|
|
|
|
|
build --config=short_logs
|
2021-11-12 12:05:33 -05:00
|
|
|
|
|
|
|
|
build --copt=-DMLIR_PYTHON_PACKAGE_PREFIX=jaxlib.mlir.
|
|
|
|
|
|
2022-05-17 13:48:17 -07:00
|
|
|
# Later Bazel flag values override earlier values; if CUDA/ROCM/TPU are enabled,
|
|
|
|
|
# these values are overridden.
|
2023-02-14 21:24:27 +00:00
|
|
|
build --@xla//xla/python:enable_gpu=false
|
2023-10-24 07:07:49 -07:00
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
###########################################################################
|
|
|
|
|
|
2021-11-12 12:05:33 -05:00
|
|
|
build:posix --copt=-fvisibility=hidden
|
2021-08-10 16:17:30 -07:00
|
|
|
build:posix --copt=-Wno-sign-compare
|
2022-05-23 17:14:31 -07:00
|
|
|
build:posix --cxxopt=-std=c++17
|
|
|
|
|
build:posix --host_cxxopt=-std=c++17
|
2021-08-10 16:17:30 -07:00
|
|
|
|
|
|
|
|
build:avx_posix --copt=-mavx
|
|
|
|
|
build:avx_posix --host_copt=-mavx
|
|
|
|
|
|
|
|
|
|
build:avx_windows --copt=/arch=AVX
|
|
|
|
|
|
|
|
|
|
build:avx_linux --copt=-mavx
|
|
|
|
|
build:avx_linux --host_copt=-mavx
|
|
|
|
|
|
|
|
|
|
build:native_arch_posix --copt=-march=native
|
|
|
|
|
build:native_arch_posix --host_copt=-march=native
|
|
|
|
|
|
|
|
|
|
build:mkl_open_source_only --define=tensorflow_mkldnn_contraction_kernel=1
|
|
|
|
|
|
|
|
|
|
build:cuda --repo_env TF_NEED_CUDA=1
|
2023-10-03 18:33:09 -07:00
|
|
|
build:cuda --repo_env TF_NCCL_USE_STUB=1
|
2022-05-02 20:03:32 +00:00
|
|
|
# "sm" means we emit only cubin, which is forward compatible within a GPU generation.
|
|
|
|
|
# "compute" means we emit both cubin and PTX, which is larger but also forward compatible to future GPU generations.
|
2023-11-28 07:15:22 -08:00
|
|
|
build:cuda --action_env TF_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,compute_90"
|
2021-08-10 16:17:30 -07:00
|
|
|
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
|
|
|
|
|
build:cuda --@local_config_cuda//:enable_cuda
|
2023-02-14 21:24:27 +00:00
|
|
|
build:cuda --@xla//xla/python:enable_gpu=true
|
2022-10-27 14:53:19 +00:00
|
|
|
build:cuda --@xla//xla/python:jax_cuda_pip_rpaths=true
|
2021-08-10 16:17:30 -07:00
|
|
|
build:cuda --define=xla_python_enable_gpu=true
|
|
|
|
|
|
2023-10-16 09:58:30 -07:00
|
|
|
# Later Bazel flag values override earlier values.
|
|
|
|
|
# TODO(jieying): remove enable_gpu and xla_python_enable_gpu from build:cuda
|
|
|
|
|
# after the pluin is released.
|
|
|
|
|
build:cuda_plugin --@xla//xla/python:enable_gpu=false
|
|
|
|
|
build:cuda_plugin --define=xla_python_enable_gpu=false
|
|
|
|
|
|
2023-10-10 10:57:56 -04:00
|
|
|
# Force the linker to set RPATH, not RUNPATH. When resolving dynamic libraries,
|
|
|
|
|
# ld.so prefers in order: RPATH, LD_LIBRARY_PATH, RUNPATH. JAX sets RPATH to
|
|
|
|
|
# point to the $ORIGIN-relative location of the pip-installed NVIDIA CUDA
|
|
|
|
|
# packages.
|
|
|
|
|
# This has pros and cons:
|
|
|
|
|
# * pro: we'll ignore other CUDA installations, which has frequently confused
|
|
|
|
|
# users in the past. By setting RPATH, we'll always use the NVIDIA pip
|
|
|
|
|
# packages if they are installed.
|
|
|
|
|
# * con: the user cannot override the CUDA installation location
|
|
|
|
|
# via LD_LIBRARY_PATH, if the nvidia-... pip packages are installed. This is
|
|
|
|
|
# acceptable, because the workaround is "remove the nvidia-..." pip packages.
|
|
|
|
|
# The list of CUDA pip packages that JAX depends on are present in setup.py.
|
|
|
|
|
build:cuda --linkopt=-Wl,--disable-new-dtags
|
|
|
|
|
|
2023-11-02 18:00:58 -07:00
|
|
|
build:cuda_clang --@local_config_cuda//:cuda_compiler=clang
|
|
|
|
|
build:cuda_clang --action_env=CLANG_CUDA_COMPILER_PATH="/usr/lib/llvm-17/bin/clang"
|
|
|
|
|
build:cuda_clang --action_env=TF_CUDA_CLANG="1"
|
|
|
|
|
# Disable clang extention 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.
|
|
|
|
|
build:cuda_clang --copt=-Wno-gnu-offsetof-extensions
|
|
|
|
|
# Disable clang extention that rejects unknown arguments.
|
|
|
|
|
build:cuda_clang --copt=-Qunused-arguments
|
|
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
build:rocm --crosstool_top=@local_config_rocm//crosstool:toolchain
|
|
|
|
|
build:rocm --define=using_rocm=true --define=using_rocm_hipcc=true
|
2023-02-14 21:24:27 +00:00
|
|
|
build:rocm --@xla//xla/python:enable_gpu=true
|
2021-08-10 16:17:30 -07:00
|
|
|
build:rocm --define=xla_python_enable_gpu=true
|
|
|
|
|
build:rocm --repo_env TF_NEED_ROCM=1
|
2022-10-27 16:57:06 +00:00
|
|
|
build:rocm --action_env TF_ROCM_AMDGPU_TARGETS="gfx900,gfx906,gfx908,gfx90a,gfx1030"
|
2021-08-10 16:17:30 -07:00
|
|
|
|
|
|
|
|
build:nonccl --define=no_nccl_support=true
|
|
|
|
|
|
2023-09-07 17:36:04 -04:00
|
|
|
# Windows has a relatively short command line limit, which JAX has begun to hit.
|
|
|
|
|
# See https://docs.bazel.build/versions/main/windows.html
|
|
|
|
|
build:windows --features=compiler_param_file
|
|
|
|
|
build:windows --features=archive_param_file
|
|
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
# Tensorflow uses M_* math constants that only get defined by MSVC headers if
|
|
|
|
|
# _USE_MATH_DEFINES is defined.
|
|
|
|
|
build:windows --copt=/D_USE_MATH_DEFINES
|
|
|
|
|
build:windows --host_copt=/D_USE_MATH_DEFINES
|
|
|
|
|
# Make sure to include as little of windows.h as possible
|
|
|
|
|
build:windows --copt=-DWIN32_LEAN_AND_MEAN
|
|
|
|
|
build:windows --host_copt=-DWIN32_LEAN_AND_MEAN
|
|
|
|
|
build:windows --copt=-DNOGDI
|
|
|
|
|
build:windows --host_copt=-DNOGDI
|
|
|
|
|
# https://devblogs.microsoft.com/cppblog/announcing-full-support-for-a-c-c-conformant-preprocessor-in-msvc/
|
|
|
|
|
# otherwise, there will be some compiling error due to preprocessing.
|
|
|
|
|
build:windows --copt=/Zc:preprocessor
|
2022-05-23 17:14:31 -07:00
|
|
|
build:windows --cxxopt=/std:c++17
|
|
|
|
|
build:windows --host_cxxopt=/std:c++17
|
2021-08-10 16:17:30 -07:00
|
|
|
# Generate PDB files, to generate useful PDBs, in opt compilation_mode
|
|
|
|
|
# --copt /Z7 is needed.
|
|
|
|
|
build:windows --linkopt=/DEBUG
|
|
|
|
|
build:windows --host_linkopt=/DEBUG
|
|
|
|
|
build:windows --linkopt=/OPT:REF
|
|
|
|
|
build:windows --host_linkopt=/OPT:REF
|
|
|
|
|
build:windows --linkopt=/OPT:ICF
|
|
|
|
|
build:windows --host_linkopt=/OPT:ICF
|
2022-09-21 17:08:13 +08:00
|
|
|
build:windows --incompatible_strict_action_env=true
|
2021-08-10 16:17:30 -07:00
|
|
|
|
|
|
|
|
build:linux --config=posix
|
2022-06-25 21:19:19 +00:00
|
|
|
build:linux --copt=-Wno-unknown-warning-option
|
2021-08-10 16:17:30 -07:00
|
|
|
# Workaround for gcc 10+ warnings related to upb.
|
|
|
|
|
# See https://github.com/tensorflow/tensorflow/issues/39467
|
|
|
|
|
build:linux --copt=-Wno-stringop-truncation
|
2021-12-14 14:27:19 -05:00
|
|
|
build:linux --copt=-Wno-array-parameter
|
2021-08-10 16:17:30 -07:00
|
|
|
|
|
|
|
|
build:macos --config=posix
|
|
|
|
|
|
|
|
|
|
# Suppress all warning messages.
|
|
|
|
|
build:short_logs --output_filter=DONT_MATCH_ANYTHING
|
|
|
|
|
|
|
|
|
|
#########################################################################
|
|
|
|
|
# RBE config options below.
|
|
|
|
|
# Flag to enable remote config
|
|
|
|
|
common --experimental_repo_remote_exec
|
|
|
|
|
|
|
|
|
|
build:rbe --repo_env=BAZEL_DO_NOT_DETECT_CPP_TOOLCHAIN=1
|
|
|
|
|
build:rbe --google_default_credentials
|
|
|
|
|
build:rbe --bes_backend=buildeventservice.googleapis.com
|
|
|
|
|
build:rbe --bes_results_url="https://source.cloud.google.com/results/invocations"
|
|
|
|
|
build:rbe --bes_timeout=600s
|
|
|
|
|
build:rbe --define=EXECUTOR=remote
|
|
|
|
|
build:rbe --flaky_test_attempts=3
|
|
|
|
|
build:rbe --jobs=200
|
|
|
|
|
build:rbe --remote_executor=grpcs://remotebuildexecution.googleapis.com
|
|
|
|
|
build:rbe --remote_timeout=3600
|
|
|
|
|
build:rbe --spawn_strategy=remote,worker,standalone,local
|
|
|
|
|
test:rbe --test_env=USER=anon
|
|
|
|
|
# Attempt to minimize the amount of data transfer between bazel and the remote
|
|
|
|
|
# workers:
|
|
|
|
|
build:rbe --remote_download_toplevel
|
|
|
|
|
|
|
|
|
|
build:rbe_linux --config=rbe
|
|
|
|
|
build:rbe_linux --action_env=PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin"
|
|
|
|
|
build:rbe_linux --host_javabase=@bazel_toolchains//configs/ubuntu16_04_clang/1.1:jdk8
|
|
|
|
|
build:rbe_linux --javabase=@bazel_toolchains//configs/ubuntu16_04_clang/1.1:jdk8
|
|
|
|
|
build:rbe_linux --host_java_toolchain=@bazel_tools//tools/jdk:toolchain_hostjdk8
|
|
|
|
|
build:rbe_linux --java_toolchain=@bazel_tools//tools/jdk:toolchain_hostjdk8
|
|
|
|
|
|
|
|
|
|
# Non-rbe settings we should include because we do not run configure
|
|
|
|
|
build:rbe_linux --config=avx_linux
|
|
|
|
|
build:rbe_linux --linkopt=-lrt
|
|
|
|
|
build:rbe_linux --host_linkopt=-lrt
|
|
|
|
|
build:rbe_linux --linkopt=-lm
|
|
|
|
|
build:rbe_linux --host_linkopt=-lm
|
|
|
|
|
|
|
|
|
|
# Use the GPU toolchain until the CPU one is ready.
|
|
|
|
|
# https://github.com/bazelbuild/bazel/issues/13623
|
2021-08-17 11:57:18 -07:00
|
|
|
build:rbe_cpu_linux_base --config=rbe_linux
|
2023-11-20 08:23:00 -08:00
|
|
|
build:rbe_cpu_linux_base --config=cuda_clang
|
|
|
|
|
build:rbe_cpu_linux_base --action_env=TF_NVCC_CLANG="1"
|
2024-01-08 01:23:55 -08:00
|
|
|
build:rbe_cpu_linux_base --host_crosstool_top="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_cuda//crosstool:toolchain"
|
|
|
|
|
build:rbe_cpu_linux_base --crosstool_top="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_cuda//crosstool:toolchain"
|
|
|
|
|
build:rbe_cpu_linux_base --extra_toolchains="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_cuda//crosstool:toolchain-linux-x86_64"
|
|
|
|
|
build:rbe_cpu_linux_base --extra_execution_platforms="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_platform//:platform"
|
|
|
|
|
build:rbe_cpu_linux_base --host_platform="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_platform//:platform"
|
|
|
|
|
build:rbe_cpu_linux_base --platforms="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_platform//:platform"
|
|
|
|
|
|
|
|
|
|
build:rbe_cpu_linux_py3.9 --config=rbe_cpu_linux_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_python3.9"
|
2023-12-05 09:41:40 -08:00
|
|
|
build:rbe_cpu_linux_py3.9 --python_path="/usr/local/bin/python3.9"
|
2024-01-08 01:23:55 -08:00
|
|
|
build:rbe_cpu_linux_py3.10 --config=rbe_cpu_linux_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_python3.10"
|
2023-12-05 09:41:40 -08:00
|
|
|
build:rbe_cpu_linux_py3.10 --python_path="/usr/local/bin/python3.10"
|
2024-01-08 01:23:55 -08:00
|
|
|
build:rbe_cpu_linux_py3.11 --config=rbe_cpu_linux_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_python3.11"
|
2023-12-05 09:41:40 -08:00
|
|
|
build:rbe_cpu_linux_py3.11 --python_path="/usr/local/bin/python3.11"
|
2024-01-08 01:23:55 -08:00
|
|
|
build:rbe_cpu_linux_py3.12 --config=rbe_cpu_linux_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_python3.12"
|
2023-12-05 09:41:40 -08:00
|
|
|
build:rbe_cpu_linux_py3.12 --python_path="/usr/local/bin/python3.12"
|
2021-08-10 16:17:30 -07:00
|
|
|
|
|
|
|
|
build:rbe_linux_cuda_base --config=rbe_linux
|
|
|
|
|
build:rbe_linux_cuda_base --config=cuda
|
|
|
|
|
build:rbe_linux_cuda_base --repo_env=REMOTE_GPU_TESTING=1
|
|
|
|
|
|
2022-11-18 16:54:58 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --config=rbe_linux_cuda_base
|
2023-11-20 08:23:00 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --config=cuda_clang
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --action_env=TF_NVCC_CLANG="1"
|
2022-11-18 16:54:58 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --action_env=TF_CUDA_VERSION=11
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --action_env=TF_CUDNN_VERSION=8
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --action_env=CUDA_TOOLKIT_PATH="/usr/local/cuda-11.8"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --action_env=LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/tensorrt/lib"
|
2023-11-20 08:23:00 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --host_crosstool_top="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_cuda//crosstool:toolchain"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --crosstool_top="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_cuda//crosstool:toolchain"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --extra_toolchains="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_cuda//crosstool:toolchain-linux-x86_64"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --extra_execution_platforms="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_platform//:platform"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --host_platform="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_platform//:platform"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --platforms="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_platform//:platform"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --repo_env=TF_CUDA_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_cuda"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_base --repo_env=TF_NCCL_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_nccl"
|
|
|
|
|
build:rbe_linux_cuda11.8_nvcc_py3.9 --config=rbe_linux_cuda11.8_nvcc_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_python3.9"
|
2022-11-18 16:54:58 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_py3.9 --python_path="/usr/local/bin/python3.9"
|
2023-11-20 08:23:00 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_py3.10 --config=rbe_linux_cuda11.8_nvcc_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_python3.10"
|
2022-11-18 16:54:58 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_py3.10 --python_path="/usr/local/bin/python3.10"
|
2023-11-20 08:23:00 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_py3.11 --config=rbe_linux_cuda11.8_nvcc_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_python3.11"
|
2022-11-18 16:54:58 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_py3.11 --python_path="/usr/local/bin/python3.11"
|
2023-11-20 08:23:00 -08:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_py3.12 --config=rbe_linux_cuda11.8_nvcc_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda11.8-cudnn8.6-tensorrt8.4_config_python3.12"
|
2023-09-21 07:55:27 -07:00
|
|
|
build:rbe_linux_cuda11.8_nvcc_py3.12 --python_path="/usr/local/bin/python3.12"
|
2022-11-18 16:54:58 -08:00
|
|
|
|
2024-01-08 01:23:55 -08:00
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --config=rbe_linux_cuda_base
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --config=cuda_clang
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --action_env=TF_NVCC_CLANG="1"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --action_env=TF_CUDA_VERSION=12
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --action_env=TF_CUDNN_VERSION=8
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --action_env=CUDA_TOOLKIT_PATH="/usr/local/cuda-12"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --action_env=LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/tensorrt/lib"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --host_crosstool_top="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_cuda//crosstool:toolchain"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --crosstool_top="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_cuda//crosstool:toolchain"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --extra_toolchains="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_cuda//crosstool:toolchain-linux-x86_64"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --extra_execution_platforms="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_platform//:platform"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --host_platform="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_platform//:platform"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --platforms="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_platform//:platform"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --repo_env=TF_CUDA_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_cuda"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --repo_env=TF_NCCL_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_nccl"
|
|
|
|
|
# RBE machines have an older CUDA driver version, so we have to enable driver forward compatibility
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_base --test_env=LD_LIBRARY_PATH=/usr/local/cuda/compat
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_py3.9 --config=rbe_linux_cuda12.3_nvcc_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_python3.9"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_py3.9 --python_path="/usr/local/bin/python3.9"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_py3.10 --config=rbe_linux_cuda12.3_nvcc_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_python3.10"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_py3.10 --python_path="/usr/local/bin/python3.10"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_py3.11 --config=rbe_linux_cuda12.3_nvcc_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_python3.11"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_py3.11 --python_path="/usr/local/bin/python3.11"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_py3.12 --config=rbe_linux_cuda12.3_nvcc_base --repo_env=TF_PYTHON_CONFIG_REPO="@ubuntu20.04-clang_manylinux2014-cuda12.3-cudnn8.9_config_python3.12"
|
|
|
|
|
build:rbe_linux_cuda12.3_nvcc_py3.12 --python_path="/usr/local/bin/python3.12"
|
2023-09-25 09:37:59 -07:00
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
# These you may need to change for your own GCP project.
|
|
|
|
|
build:tensorflow_testing_rbe --project_id=tensorflow-testing
|
|
|
|
|
common:tensorflow_testing_rbe_linux --remote_instance_name=projects/tensorflow-testing/instances/default_instance
|
|
|
|
|
build:tensorflow_testing_rbe_linux --config=tensorflow_testing_rbe
|
|
|
|
|
#############################################################################
|
|
|
|
|
|
2023-02-10 21:03:21 -08:00
|
|
|
#############################################################################
|
|
|
|
|
# Some configs to make getting some forms of debug builds. In general, the
|
|
|
|
|
# codebase is only regularly built with optimizations. Use 'debug_symbols' to
|
|
|
|
|
# just get symbols for the parts of XLA/PJRT that jaxlib uses.
|
2023-10-12 17:51:35 -07:00
|
|
|
# Or try 'debug' to get a build with assertions enabled and minimal
|
2023-02-10 21:03:21 -08:00
|
|
|
# optimizations.
|
|
|
|
|
# Include these in a local .bazelrc.user file as:
|
|
|
|
|
# build --config=debug_symbols
|
|
|
|
|
# Or:
|
|
|
|
|
# build --config=debug
|
|
|
|
|
#
|
|
|
|
|
# Additional files can be opted in for debug symbols by adding patterns
|
|
|
|
|
# to a per_file_copt similar to below.
|
|
|
|
|
#############################################################################
|
|
|
|
|
|
|
|
|
|
build:debug_symbols --strip=never --per_file_copt="xla/pjrt|xla/python@-g3"
|
|
|
|
|
build:debug --config debug_symbols -c fastbuild
|
|
|
|
|
|
2021-08-10 16:17:30 -07:00
|
|
|
# Load `.jax_configure.bazelrc` file written by build.py
|
|
|
|
|
try-import %workspace%/.jax_configure.bazelrc
|
2022-08-14 15:46:12 -04:00
|
|
|
|
|
|
|
|
# Load rc file with user-specific options.
|
|
|
|
|
try-import %workspace%/.bazelrc.user
|
