Skip to content

Commit 4c7092b

Browse files
Merge branch 'master' into aimran/onednn3.7
2 parents 1217a69 + e5f4bb8 commit 4c7092b

5,297 files changed

Lines changed: 224301 additions & 205168 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

.bazelrc

Lines changed: 29 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,10 @@ build --config=short_logs
125125
# TODO(mihaimaruseac): Document this option or remove if no longer needed
126126
build --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
130134
common --experimental_cc_shared_library
@@ -159,15 +163,19 @@ build:android --host_crosstool_top=@bazel_tools//tools/cpp:toolchain
159163
build:android_arm --config=android
160164
build:android_arm --cpu=armeabi-v7a
161165
build:android_arm --fat_apk_cpu=armeabi-v7a
166+
build:android_arm --platforms=@org_tensorflow//tensorflow/tools/toolchains/android:armeabi-v7a
162167
build:android_arm64 --config=android
163168
build:android_arm64 --cpu=arm64-v8a
164169
build:android_arm64 --fat_apk_cpu=arm64-v8a
170+
build:android_arm64 --platforms=@org_tensorflow//tensorflow/tools/toolchains/android:arm64-v8a
165171
build:android_x86 --config=android
166172
build:android_x86 --cpu=x86
167173
build:android_x86 --fat_apk_cpu=x86
174+
build:android_x86 --platforms=@org_tensorflow//tensorflow/tools/toolchains/android:x86
168175
build:android_x86_64 --config=android
169176
build:android_x86_64 --cpu=x86_64
170177
build: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.
201209
build:macos_arm64 --cpu=darwin_arm64
202210
build: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.
205214
build:ios --apple_platform_type=ios
@@ -208,14 +217,19 @@ build:ios --copt=-Wno-c++11-narrowing
208217
build:ios --config=apple-toolchain
209218
build:ios_armv7 --config=ios
210219
build:ios_armv7 --cpu=ios_armv7
220+
build:ios_armv7 --platforms=@org_tensorflow//tensorflow/tools/toolchains/ios:ios_armv7
211221
build:ios_arm64 --config=ios
212222
build:ios_arm64 --cpu=ios_arm64
223+
build:ios_arm64 --platforms=@build_bazel_apple_support//configs/platforms:ios_arm64
213224
build:ios_arm64e --config=ios
214225
build:ios_arm64e --cpu=ios_arm64e
226+
build:ios_arm64e --platforms=@build_bazel_apple_support//configs/platforms:ios_arm64e
215227
build:ios_sim_arm64 --config=ios
216228
build:ios_sim_arm64 --cpu=ios_sim_arm64
229+
build:ios_sim_arm64 --platforms=@build_bazel_apple_support//configs/platforms:ios_sim_arm64
217230
build:ios_x86_64 --config=ios
218231
build:ios_x86_64 --cpu=ios_x86_64
232+
build:ios_x86_64 --platforms=@build_bazel_apple_support//configs/platforms:ios_x86_64
219233
build:ios_fat --config=ios
220234
build:ios_fat --ios_multi_cpus=armv7,arm64,i386,x86_64
221235

@@ -252,13 +266,15 @@ build:mkl_aarch64 -c opt
252266
build:mkl_aarch64_threadpool --define=build_with_mkl_aarch64=true
253267
build: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.
256274
build:cuda --repo_env TF_NEED_CUDA=1
257275
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
258276
build: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.
263279
build: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.
290306
build: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
293308
build:cuda_clang_official --action_env=CLANG_CUDA_COMPILER_PATH="/usr/lib/llvm-18/bin/clang"
294309
build: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
428443
build:linux --define=PREFIX=/usr
429-
build:linux --define=LIBDIR=$(PREFIX)/lib
430-
build:linux --define=INCLUDEDIR=$(PREFIX)/include
431444
build:linux --define=PROTOBUF_INCLUDE_PATH=$(PREFIX)/include
432445
build:macos --define=PREFIX=/usr
433-
build:macos --define=LIBDIR=$(PREFIX)/lib
434-
build:macos --define=INCLUDEDIR=$(PREFIX)/include
435446
build: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.
593604
common: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.
596613
build:rbe_linux_cpu_old --config=rbe_linux
597614
build: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

606623
build:rbe_linux_cuda --config=cuda_clang_official
607624
build: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
609627
build:rbe_linux_cuda --repo_env=REMOTE_GPU_TESTING=1
610628

@@ -632,8 +650,10 @@ build:elinux --crosstool_top=@local_config_embedded_arm//:toolchain
632650
build:elinux --host_crosstool_top=@bazel_tools//tools/cpp:toolchain
633651
build:elinux_aarch64 --config=elinux
634652
build:elinux_aarch64 --cpu=aarch64
653+
build:elinux_aarch64 --platforms=@org_tensorflow//tensorflow/tools/toolchains/linux:linux_aarch64
635654
build:elinux_armhf --config=elinux
636655
build:elinux_armhf --cpu=armhf
656+
build:elinux_armhf --platforms=@org_tensorflow//tensorflow/tools/toolchains/linux:linux_armhf
637657
build:elinux_armhf --copt -mfp16-format=ieee
638658

639659
# Config-specific options should come above this line.

.github/workflows/osv-scanner-scheduled.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ permissions:
2828
jobs:
2929
scan-scheduled:
3030
if: github.repository == 'tensorflow/tensorflow'
31-
uses: "google/osv-scanner-action/.github/workflows/osv-scanner-reusable.yml@v1.9.2"
31+
uses: "google/osv-scanner-action/.github/workflows/osv-scanner-reusable.yml@v2.0.1"
3232
with:
3333
scan-args: |-
3434
--lockfile=requirements.txt:./requirements_lock_3_9.txt

.github/workflows/pylint-presubmit.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ jobs:
3838
run: |
3939
echo Changed files: ${{ steps.get_file_changes.outputs.files }}
4040
- name: Set up Python 3.9
41-
uses: actions/setup-python@42375524e23c412d93fb67b49958b491fce71c38 # v5.4.0
41+
uses: actions/setup-python@a26af69be951a213d495a4c3e4e4022e16d87065 # v5.6.0
4242
with:
4343
python-version: "3.9"
4444
- name: Install Python dependencies

.github/workflows/release-branch-cherrypick.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ jobs:
5858
echo "SHORTSHA=$(git log -1 ${{ github.event.inputs.git_commit }} --format="%h")" >> "$GITHUB_OUTPUT"
5959
echo "TITLE=$(git log -1 ${{ github.event.inputs.git_commit }} --format="%s")" >> "$GITHUB_OUTPUT"
6060
- name: Create Pull Request with changes
61-
uses: peter-evans/create-pull-request@dd2324fc52d5d43c699a5636bcf19fceaa70c284 # v7.0.7
61+
uses: peter-evans/create-pull-request@271a8d0340265f705b14b6d32b9829c1cb33d45e # v7.0.8
6262
with:
6363
title: '${{ github.event.inputs.release_branch }} cherry-pick: ${{ steps.cherrypick.outputs.SHORTSHA }} "${{ steps.cherrypick.outputs.TITLE }}"'
6464
committer: TensorFlow Release Automation <jenkins@tensorflow.org>

.github/workflows/scorecards-analysis.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ jobs:
5555
# Upload the results as artifacts (optional). Commenting out will disable uploads of run results in SARIF
5656
# format to the repository Actions tab.
5757
- name: "Upload artifact"
58-
uses: actions/upload-artifact@4cec3d8aa04e39d1a68397de0c4cd6fb9dce8ec1 # v4.6.1
58+
uses: actions/upload-artifact@ea165f8d65b6e75b540449e92b4886f43607fa02 # v4.6.2
5959
with:
6060
name: SARIF file
6161
path: results.sarif
@@ -64,6 +64,6 @@ jobs:
6464
# Upload the results to GitHub's code scanning dashboard (optional).
6565
# Commenting out will disable upload of results to your repo's Code Scanning dashboard
6666
- name: "Upload to code-scanning"
67-
uses: github/codeql-action/upload-sarif@b56ba49b26e50535fa1e7f7db0f4f7b4bf65d80d # v3.28.10
67+
uses: github/codeql-action/upload-sarif@28deaeda66b76a05916b6923827895f2b14ab387 # v3.28.16
6868
with:
6969
sarif_file: results.sarif

.github/workflows/update-rbe.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ jobs:
130130
map sigbuild-r2.17-clang-python3.11 2.17-python3.11
131131
map sigbuild-r2.17-clang-python3.12 2.17-python3.12
132132
- name: Create Pull Request with changes
133-
uses: peter-evans/create-pull-request@dd2324fc52d5d43c699a5636bcf19fceaa70c284 # v7.0.7
133+
uses: peter-evans/create-pull-request@271a8d0340265f705b14b6d32b9829c1cb33d45e # v7.0.8
134134
with:
135135
title: Update the RBE images to the latest container versions
136136
committer: TensorFlow Release Automation <jenkins@tensorflow.org>

README.md

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,6 @@
1111
[![Fuzzing Status](https://oss-fuzz-build-logs.storage.googleapis.com/badges/tensorflow-py.svg)](https://bugs.chromium.org/p/oss-fuzz/issues/list?sort=-opened&can=1&q=proj:tensorflow-py)
1212
[![OSSRank](https://shields.io/endpoint?url=https://ossrank.com/shield/44)](https://ossrank.com/p/44)
1313
[![Contributor Covenant](https://img.shields.io/badge/Contributor%20Covenant-v1.4%20adopted-ff69b4.svg)](CODE_OF_CONDUCT.md)
14-
[![TF Official Continuous](https://tensorflow.github.io/build/TF%20Official%20Continuous.svg)](https://tensorflow.github.io/build#TF%20Official%20Continuous)
15-
[![TF Official Nightly](https://tensorflow.github.io/build/TF%20Official%20Nightly.svg)](https://tensorflow.github.io/build#TF%20Official%20Nightly)
1614

1715
**`Documentation`** |
1816
------------------- |

RELEASE.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3234,7 +3234,7 @@ This release introduces several vulnerability fixes:
32343234

32353235
* Keras been split into a separate PIP package (`keras`), and its code has
32363236
been moved to the GitHub
3237-
repository[keras-team/keras](http://github.com/keras-team/keras). The
3237+
repository[keras-team/keras](https://github.com/keras-team/keras). The
32383238
API endpoints for `tf.keras` stay unchanged, but are now backed by the
32393239
`keras` PIP package. The existing code in tensorflow/python/keras is a
32403240
staled copy and will be removed in future release (2.7). Please remove
@@ -10260,7 +10260,7 @@ answered questions, and were part of inspiring discussions.
1026010260
## Major Features And Improvements
1026110261

1026210262
* `tf.keras` is now part of the core TensorFlow API.
10263-
* [`tf.data`](http://tensorflow.org/guide/data) is now part of the core
10263+
* [`tf.data`](https://tensorflow.org/guide/data) is now part of the core
1026410264
TensorFlow API.
1026510265
* The API is now subject to backwards compatibility guarantees.
1026610266
* For a guide to migrating from the `tf.contrib.data` API, see the

WORKSPACE

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,13 @@ http_archive(
1313
url = "https://github.com/bazelbuild/rules_java/releases/download/5.3.5/rules_java-5.3.5.tar.gz",
1414
)
1515

16+
http_archive(
17+
name = "rules_shell",
18+
sha256 = "bc61ef94facc78e20a645726f64756e5e285a045037c7a61f65af2941f4c25e1",
19+
strip_prefix = "rules_shell-0.4.1",
20+
url = "https://github.com/bazelbuild/rules_shell/releases/download/v0.4.1/rules_shell-v0.4.1.tar.gz",
21+
)
22+
1623
# Initialize the TensorFlow repository and all dependencies.
1724
#
1825
# The cascade of load() statements and tf_workspace?() calls works around the
@@ -23,6 +30,12 @@ load("@//tensorflow:workspace3.bzl", "tf_workspace3")
2330

2431
tf_workspace3()
2532

33+
load("@rules_shell//shell:repositories.bzl", "rules_shell_dependencies", "rules_shell_toolchains")
34+
35+
rules_shell_dependencies()
36+
37+
rules_shell_toolchains()
38+
2639
# Initialize hermetic Python
2740
load("@local_xla//third_party/py:python_init_rules.bzl", "python_init_rules")
2841

@@ -43,6 +56,7 @@ python_init_repositories(
4356
"3.10": "//:requirements_lock_3_10.txt",
4457
"3.11": "//:requirements_lock_3_11.txt",
4558
"3.12": "//:requirements_lock_3_12.txt",
59+
"3.13": "//:requirements_lock_3_13.txt",
4660
},
4761
)
4862

ci/official/containers/linux_arm64/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
################################################################################
2-
FROM ubuntu:20.04@sha256:8e5c4f0285ecbb4ead070431d29b576a530d3166df73ec44affc1cd27555141b as builder
2+
FROM ubuntu:20.04@sha256:8feb4d8ca5354def3d8fce243717141ce31e2c428701f6682bd2fafe15388214 as builder
33
################################################################################
44

55
# Install devtoolset build dependencies

0 commit comments

Comments
 (0)