Skip to content

Commit e7cf8a9

Browse files
rtg0795tf-text-github-robot
authored andcommitted
Add Hermetic CUDA rules.
PiperOrigin-RevId: 664903046
1 parent 748c121 commit e7cf8a9

File tree

1 file changed

+54
-7
lines changed

1 file changed

+54
-7
lines changed

WORKSPACE

+54-7
Original file line numberDiff line numberDiff line change
@@ -56,10 +56,10 @@ http_archive(
5656

5757
http_archive(
5858
name = "org_tensorflow",
59-
strip_prefix = "tensorflow-9d2929b69fdddc1e8d5de70a706556b15d8ed37b",
60-
sha256 = "a4f965340ea11d49c8897df59a24822c2b367e4cebb5908b7ca8c3b607097af9",
59+
strip_prefix = "tensorflow-1ad7a2160315a64dd2a5cbde380e9478b671d043",
60+
sha256 = "2caf5f221197fa817f41f8b71a14adda3261e2ecd1b45357f5af824530b37267",
6161
urls = [
62-
"https://github.com/tensorflow/tensorflow/archive/9d2929b69fdddc1e8d5de70a706556b15d8ed37b.zip"
62+
"https://github.com/tensorflow/tensorflow/archive/1ad7a2160315a64dd2a5cbde380e9478b671d043.zip"
6363
],
6464
)
6565

@@ -75,11 +75,11 @@ http_archive(
7575
http_archive(
7676
name = "pybind11",
7777
urls = [
78-
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/pybind/pybind11/archive/v2.10.0.tar.gz",
79-
"https://github.com/pybind/pybind11/archive/v2.10.0.tar.gz",
78+
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/pybind/pybind11/archive/v2.13.4.tar.gz",
79+
"https://github.com/pybind/pybind11/archive/v2.13.4.tar.gz",
8080
],
81-
sha256 = "eacf582fa8f696227988d08cfc46121770823839fe9e301a20fbce67e7cd70ec",
82-
strip_prefix = "pybind11-2.10.0",
81+
sha256 = "efc901aa0aab439a3fea6efeaf930b5a349fb06394bf845c64ce15a9cf8f0240",
82+
strip_prefix = "pybind11-2.13.4",
8383
build_file = "//third_party/pybind11:BUILD.bzl",
8484
)
8585

@@ -141,3 +141,50 @@ load("@org_tensorflow//third_party/android:android_configure.bzl", "android_conf
141141
android_configure(name="local_config_android")
142142
load("@local_config_android//:android.bzl", "android_workspace")
143143
android_workspace()
144+
145+
load(
146+
"@local_tsl//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl",
147+
"cuda_json_init_repository",
148+
)
149+
150+
cuda_json_init_repository()
151+
152+
load(
153+
"@cuda_redist_json//:distributions.bzl",
154+
"CUDA_REDISTRIBUTIONS",
155+
"CUDNN_REDISTRIBUTIONS",
156+
)
157+
load(
158+
"@local_tsl//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl",
159+
"cuda_redist_init_repositories",
160+
"cudnn_redist_init_repository",
161+
)
162+
163+
cuda_redist_init_repositories(
164+
cuda_redistributions = CUDA_REDISTRIBUTIONS,
165+
)
166+
167+
cudnn_redist_init_repository(
168+
cudnn_redistributions = CUDNN_REDISTRIBUTIONS,
169+
)
170+
171+
load(
172+
"@local_tsl//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
173+
"cuda_configure",
174+
)
175+
176+
cuda_configure(name = "local_config_cuda")
177+
178+
load(
179+
"@local_tsl//third_party/nccl/hermetic:nccl_redist_init_repository.bzl",
180+
"nccl_redist_init_repository",
181+
)
182+
183+
nccl_redist_init_repository()
184+
185+
load(
186+
"@local_tsl//third_party/nccl/hermetic:nccl_configure.bzl",
187+
"nccl_configure",
188+
)
189+
190+
nccl_configure(name = "local_config_nccl")

0 commit comments

Comments
 (0)