@@ -125,6 +125,10 @@ build --config=short_logs
125125# TODO(mihaimaruseac): Document this option or remove if no longer needed
126126build --config=v2
127127
128+ # Precompiling results in some action conflicts. Disable it for now until
129+ # the problematic targets are fixed.
130+ build --@rules_python//python/config_settings:precompile=force_disabled
131+
128132# TF now has `cc_shared_library` targets, so it needs the experimental flag
129133# TODO(rostam): Remove when `cc_shared_library` is enabled by default
130134common --experimental_cc_shared_library
@@ -159,15 +163,19 @@ build:android --host_crosstool_top=@bazel_tools//tools/cpp:toolchain
159163build:android_arm --config=android
160164build:android_arm --cpu=armeabi-v7a
161165build:android_arm --fat_apk_cpu=armeabi-v7a
166+ build:android_arm --platforms=@org_tensorflow//tensorflow/tools/toolchains/android:armeabi-v7a
162167build:android_arm64 --config=android
163168build:android_arm64 --cpu=arm64-v8a
164169build:android_arm64 --fat_apk_cpu=arm64-v8a
170+ build:android_arm64 --platforms=@org_tensorflow//tensorflow/tools/toolchains/android:arm64-v8a
165171build:android_x86 --config=android
166172build:android_x86 --cpu=x86
167173build:android_x86 --fat_apk_cpu=x86
174+ build:android_x86 --platforms=@org_tensorflow//tensorflow/tools/toolchains/android:x86
168175build:android_x86_64 --config=android
169176build:android_x86_64 --cpu=x86_64
170177build:android_x86_64 --fat_apk_cpu=x86_64
178+ build:android_x86_64 --platforms=@org_tensorflow//tensorflow/tools/toolchains/android:x86_64
171179
172180# Build everything statically for Android since all static libs are later
173181# bundled together into a single .so for deployment.
@@ -200,6 +208,7 @@ build:apple-toolchain --host_crosstool_top=@local_config_apple_cc//:toolchain
200208# Settings for MacOS on ARM CPUs.
201209build:macos_arm64 --cpu=darwin_arm64
202210build:macos_arm64 --macos_minimum_os=11.0
211+ build:macos_arm64 --platforms=@build_bazel_apple_support//configs/platforms:darwin_arm64
203212
204213# iOS configs for each architecture and the fat binary builds.
205214build:ios --apple_platform_type=ios
@@ -208,14 +217,19 @@ build:ios --copt=-Wno-c++11-narrowing
208217build:ios --config=apple-toolchain
209218build:ios_armv7 --config=ios
210219build:ios_armv7 --cpu=ios_armv7
220+ build:ios_armv7 --platforms=@org_tensorflow//tensorflow/tools/toolchains/ios:ios_armv7
211221build:ios_arm64 --config=ios
212222build:ios_arm64 --cpu=ios_arm64
223+ build:ios_arm64 --platforms=@build_bazel_apple_support//configs/platforms:ios_arm64
213224build:ios_arm64e --config=ios
214225build:ios_arm64e --cpu=ios_arm64e
226+ build:ios_arm64e --platforms=@build_bazel_apple_support//configs/platforms:ios_arm64e
215227build:ios_sim_arm64 --config=ios
216228build:ios_sim_arm64 --cpu=ios_sim_arm64
229+ build:ios_sim_arm64 --platforms=@build_bazel_apple_support//configs/platforms:ios_sim_arm64
217230build:ios_x86_64 --config=ios
218231build:ios_x86_64 --cpu=ios_x86_64
232+ build:ios_x86_64 --platforms=@build_bazel_apple_support//configs/platforms:ios_x86_64
219233build:ios_fat --config=ios
220234build:ios_fat --ios_multi_cpus=armv7,arm64,i386,x86_64
221235
@@ -252,13 +266,15 @@ build:mkl_aarch64 -c opt
252266build:mkl_aarch64_threadpool --define=build_with_mkl_aarch64=true
253267build:mkl_aarch64_threadpool -c opt
254268
269+ # Default CUDA and CUDNN versions.
270+ build:cuda_version --repo_env=HERMETIC_CUDA_VERSION="12.5.1"
271+ build:cuda_version --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"
272+
255273# CUDA: This config refers to building CUDA op kernels with nvcc.
256274build:cuda --repo_env TF_NEED_CUDA=1
257275build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
258276build:cuda --@local_config_cuda//:enable_cuda
259- # Default CUDA and CUDNN versions.
260- build:cuda --repo_env=HERMETIC_CUDA_VERSION="12.5.1"
261- build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"
277+ build:cuda --config=cuda_version
262278# This flag is needed to include CUDA libraries.
263279build:cuda --@local_config_cuda//cuda:include_cuda_libs=true
264280
@@ -288,8 +304,7 @@ build:cuda_clang --linkopt="-lm"
288304
289305# Set up compilation CUDA version and paths and use the CUDA Clang toolchain.
290306build:cuda_clang_official --config=cuda_clang
291- build:cuda_clang_official --repo_env=HERMETIC_CUDA_VERSION="12.5.1"
292- build:cuda_clang_official --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"
307+ build:cuda_clang_official --config=cuda_version
293308build:cuda_clang_official --action_env=CLANG_CUDA_COMPILER_PATH="/usr/lib/llvm-18/bin/clang"
294309build:cuda_clang_official --crosstool_top="@local_config_cuda//crosstool:toolchain"
295310
@@ -426,12 +441,8 @@ build:windows --dynamic_mode=off
426441
427442# Default paths for TF_SYSTEM_LIBS
428443build:linux --define=PREFIX=/usr
429- build:linux --define=LIBDIR=$(PREFIX)/lib
430- build:linux --define=INCLUDEDIR=$(PREFIX)/include
431444build:linux --define=PROTOBUF_INCLUDE_PATH=$(PREFIX)/include
432445build:macos --define=PREFIX=/usr
433- build:macos --define=LIBDIR=$(PREFIX)/lib
434- build:macos --define=INCLUDEDIR=$(PREFIX)/include
435446build:macos --define=PROTOBUF_INCLUDE_PATH=$(PREFIX)/include
436447# TF_SYSTEM_LIBS do not work on windows.
437448
@@ -592,6 +603,12 @@ build:rbe_linux_cpu --python_path="/usr/bin/python3"
592603# These you may need to change for your own GCP project.
593604common:rbe_linux_cpu --remote_instance_name=projects/tensorflow-testing/instances/default_instance
594605
606+ # Download CUDA/CUDNN redistributions to preserve the repositories cache between
607+ # CPU and GPU builds.
608+ # TODO(ybaturina): Uncomment when RBE is ready to support this.
609+ # build:rbe_linux_cpu --repo_env USE_CUDA_REDISTRIBUTIONS=1
610+ # build:rbe_linux_cpu --config=cuda_version
611+
595612# TODO(kanglan): Remove it after toolchain update is complete.
596613build:rbe_linux_cpu_old --config=rbe_linux
597614build:rbe_linux_cpu_old --host_crosstool_top="@ubuntu20.04-gcc9_manylinux2014-cuda11.2-cudnn8.1-tensorrt7.2_config_cuda//crosstool:toolchain"
@@ -605,6 +622,7 @@ common:rbe_linux_cpu_old --remote_instance_name=projects/tensorflow-testing/inst
605622
606623build:rbe_linux_cuda --config=cuda_clang_official
607624build:rbe_linux_cuda --config=rbe_linux_cpu
625+ build:rbe_linux_cuda --repo_env=USE_CUDA_TAR_ARCHIVE_FILES=1
608626# For Remote build execution -- GPU configuration
609627build:rbe_linux_cuda --repo_env=REMOTE_GPU_TESTING=1
610628
@@ -632,8 +650,10 @@ build:elinux --crosstool_top=@local_config_embedded_arm//:toolchain
632650build:elinux --host_crosstool_top=@bazel_tools//tools/cpp:toolchain
633651build:elinux_aarch64 --config=elinux
634652build:elinux_aarch64 --cpu=aarch64
653+ build:elinux_aarch64 --platforms=@org_tensorflow//tensorflow/tools/toolchains/linux:linux_aarch64
635654build:elinux_armhf --config=elinux
636655build:elinux_armhf --cpu=armhf
656+ build:elinux_armhf --platforms=@org_tensorflow//tensorflow/tools/toolchains/linux:linux_armhf
637657build:elinux_armhf --copt -mfp16-format=ieee
638658
639659# Config-specific options should come above this line.
0 commit comments