Update XLA to version 20250204.0-6789523 and adjust Bazel module and runtime files for Bazel 8 compatibility.
This commit is contained in:
parent
a34190679b
commit
f5ab6ff2c6
16
MODULE.bazel
16
MODULE.bazel
@ -10,10 +10,10 @@ bazel_dep(name = "aspect_rules_py", version = "1.3.1")
|
||||
bazel_dep(name = "bazel_skylib", version = "1.7.1")
|
||||
bazel_dep(name = "hermetic_cc_toolchain", version = "3.1.1")
|
||||
bazel_dep(name = "libxev", version = "20252401.0-31eed4e")
|
||||
bazel_dep(name = "llvm-raw", version = "20250117.0-bf17016")
|
||||
bazel_dep(name = "llvm-raw", version = "20250201.0-386af4a")
|
||||
bazel_dep(name = "patchelf", version = "0.18.0")
|
||||
bazel_dep(name = "pcre2", version = "10.43")
|
||||
bazel_dep(name = "platforms", version = "0.0.10")
|
||||
bazel_dep(name = "platforms", version = "0.0.11")
|
||||
bazel_dep(name = "protobuf", version = "29.2")
|
||||
bazel_dep(name = "rules_cc", version = "0.1.0")
|
||||
bazel_dep(name = "rules_distroless", version = "0.5.1")
|
||||
@ -23,10 +23,10 @@ bazel_dep(name = "rules_rust", version = "0.57.1")
|
||||
bazel_dep(name = "rules_uv", version = "0.53.0")
|
||||
bazel_dep(name = "rules_zig", version = "20240913.0-1957d05")
|
||||
bazel_dep(name = "sentencepiece", version = "20240618.0-d7ace0a")
|
||||
bazel_dep(name = "stablehlo", version = "20250117.0-c125b32")
|
||||
bazel_dep(name = "stablehlo", version = "20250201.0-7775e3e")
|
||||
bazel_dep(name = "toolchains_protoc", version = "0.3.7")
|
||||
bazel_dep(name = "with_cfg.bzl", version = "0.8.0")
|
||||
bazel_dep(name = "xla", version = "20250123.0-cc075be")
|
||||
bazel_dep(name = "xla", version = "20250204-0-6789523")
|
||||
bazel_dep(name = "zig-protobuf", version = "20240722.0-c644d11")
|
||||
bazel_dep(name = "zig-yaml", version = "20240903.0-83d5fdf")
|
||||
|
||||
@ -42,6 +42,7 @@ protoc.toolchain(
|
||||
version = "v29.2",
|
||||
)
|
||||
use_repo(protoc, "com_google_protobuf", "toolchains_protoc_hub")
|
||||
|
||||
register_toolchains("@toolchains_protoc_hub//:all")
|
||||
|
||||
zig = use_extension("@rules_zig//zig:extensions.bzl", "zig")
|
||||
@ -51,11 +52,14 @@ zig.mirrors(urls = [
|
||||
"https://mirror.zml.ai/zig",
|
||||
])
|
||||
use_repo(zig, "zig_toolchains")
|
||||
|
||||
register_toolchains("@rules_zig//zig/target:all")
|
||||
|
||||
register_toolchains("@zig_toolchains//:all")
|
||||
|
||||
toolchains = use_extension("@hermetic_cc_toolchain//toolchain:ext.bzl", "toolchains")
|
||||
use_repo(toolchains, "zig_sdk")
|
||||
|
||||
register_toolchains(
|
||||
"@zig_sdk//toolchain:linux_amd64_gnu.2.31",
|
||||
"@zig_sdk//toolchain:linux_arm64_gnu.2.31",
|
||||
@ -88,7 +92,7 @@ cuda = use_extension("//runtimes/cuda:cuda.bzl", "cuda_packages")
|
||||
use_repo(cuda, "libpjrt_cuda")
|
||||
|
||||
rocm = use_extension("//runtimes/rocm:rocm.bzl", "rocm_packages")
|
||||
use_repo(rocm, "libpjrt_rocm")
|
||||
use_repo(rocm, "hipblaslt", "libpjrt_rocm", "rocblas")
|
||||
|
||||
tpu = use_extension("//runtimes/tpu:tpu.bzl", "tpu_packages")
|
||||
use_repo(tpu, "libpjrt_tpu")
|
||||
@ -98,6 +102,7 @@ use_repo(neuron, "aws-neuronx-collectives", "aws-neuronx-runtime-lib")
|
||||
|
||||
zls = use_extension("//third_party/zls:zls.bzl", "repo")
|
||||
use_repo(zls, "zls_aarch64-macos", "zls_x86_64-linux")
|
||||
|
||||
register_toolchains("//third_party/zls:all")
|
||||
|
||||
llvm = use_extension("@llvm-raw//utils/bazel:extension.bzl", "llvm")
|
||||
@ -132,6 +137,7 @@ rust.toolchain(
|
||||
versions = ["1.84.0"],
|
||||
)
|
||||
use_repo(rust, "rust_toolchains")
|
||||
|
||||
register_toolchains("@rust_toolchains//:all")
|
||||
|
||||
crate = use_extension("@rules_rust//crate_universe:extensions.bzl", "crate")
|
||||
|
||||
@ -29,7 +29,6 @@
|
||||
"https://bazel-registry.zml.ai/modules/bazel_features/1.20.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/bazel_features/1.21.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/bazel_features/1.22.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/bazel_features/1.23.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/bazel_features/1.3.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/bazel_features/1.4.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/bazel_features/1.9.0/MODULE.bazel": "not found",
|
||||
@ -77,14 +76,13 @@
|
||||
"https://bazel-registry.zml.ai/modules/protobuf/29.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/protobuf/29.2/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/protobuf/3.19.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/pybind11_bazel/2.11.1.bzl.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/pybind11_bazel/2.11.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/pybind11_bazel/2.12.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/re2/2023-09-01/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/re2/2024-02-01/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/re2/2024-07-02.bcr.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/re2/2024-07-02/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_android/0.1.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_apple/3.2.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_apple/3.17.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_cc/0.0.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_cc/0.0.10/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_cc/0.0.13/MODULE.bazel": "not found",
|
||||
@ -150,8 +148,8 @@
|
||||
"https://bazel-registry.zml.ai/modules/rules_rust/0.57.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_shell/0.2.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_shell/0.3.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_shell/0.4.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_swift/1.15.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_swift/1.16.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_swift/2.1.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/rules_uv/0.53.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/stardoc/0.5.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/stardoc/0.5.3/MODULE.bazel": "not found",
|
||||
@ -159,6 +157,7 @@
|
||||
"https://bazel-registry.zml.ai/modules/stardoc/0.6.2/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/stardoc/0.7.0/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/stardoc/0.7.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/swift_argument_parser/1.3.1.1/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/toolchains_protoc/0.3.7/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/upb/0.0.0-20220923-a547704/MODULE.bazel": "not found",
|
||||
"https://bazel-registry.zml.ai/modules/upb/0.0.0-20230516-61a97ef/MODULE.bazel": "not found",
|
||||
@ -201,8 +200,7 @@
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.20.0/MODULE.bazel": "8b85300b9c8594752e0721a37210e34879d23adc219ed9dc8f4104a4a1750920",
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.21.0/MODULE.bazel": "675642261665d8eea09989aa3b8afb5c37627f1be178382c320d1b46afba5e3b",
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.22.0/MODULE.bazel": "5a5571e4155373c0463f568edfbcc1fb81dde7ec9e7d980999f1f03bff8d93cc",
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.23.0/MODULE.bazel": "fd1ac84bc4e97a5a0816b7fd7d4d4f6d837b0047cf4cbd81652d616af3a6591a",
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.23.0/source.json": "c72c61b722d7c3f884994fe647afeb2ed1ae66c437f8f370753551f7b4d8be7f",
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.22.0/source.json": "c5ff5b7ca29ad2838dadc497c793e327cd903c7d650afa215d0ef98169d04646",
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.3.0/MODULE.bazel": "cdcafe83ec318cda34e02948e81d790aab8df7a929cec6f6969f13a489ccecd9",
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.4.1/MODULE.bazel": "e45b6bb2350aff3e442ae1111c555e27eac1d915e77775f6fdc4b351b758b5d7",
|
||||
"https://bcr.bazel.build/modules/bazel_features/1.9.0/MODULE.bazel": "885151d58d90d8d9c811eb75e3288c11f850e1d6b481a8c9f766adee4712358b",
|
||||
@ -261,18 +259,17 @@
|
||||
"https://bcr.bazel.build/modules/protobuf/29.2/MODULE.bazel": "5435497c190d86f79b0568698c45044df7c8d97692886cda9fe9cf9053aea712",
|
||||
"https://bcr.bazel.build/modules/protobuf/29.2/source.json": "fe7090cc34072609b26d9beafb122916dabc1d47ba61b242c26c4b06c51384ab",
|
||||
"https://bcr.bazel.build/modules/protobuf/3.19.0/MODULE.bazel": "6b5fbb433f760a99a22b18b6850ed5784ef0e9928a72668b66e4d7ccd47db9b0",
|
||||
"https://bcr.bazel.build/modules/pybind11_bazel/2.11.1.bzl.1/MODULE.bazel": "1ef2994c097ee88f8f7ae8fbf991aaefb0603b2540fe575eca14943bc9f220a6",
|
||||
"https://bcr.bazel.build/modules/pybind11_bazel/2.11.1/MODULE.bazel": "88af1c246226d87e65be78ed49ecd1e6f5e98648558c14ce99176da041dc378e",
|
||||
"https://bcr.bazel.build/modules/pybind11_bazel/2.12.0/MODULE.bazel": "e6f4c20442eaa7c90d7190d8dc539d0ab422f95c65a57cc59562170c58ae3d34",
|
||||
"https://bcr.bazel.build/modules/pybind11_bazel/2.12.0/source.json": "6900fdc8a9e95866b8c0d4ad4aba4d4236317b5c1cd04c502df3f0d33afed680",
|
||||
"https://bcr.bazel.build/modules/re2/2023-09-01/MODULE.bazel": "cb3d511531b16cfc78a225a9e2136007a48cf8a677e4264baeab57fe78a80206",
|
||||
"https://bcr.bazel.build/modules/re2/2024-02-01/MODULE.bazel": "5ed922cb8b6c110e30969695e73bd0d3159576bf17ce8ee2443a7d07bf500551",
|
||||
"https://bcr.bazel.build/modules/re2/2024-07-02.bcr.1/MODULE.bazel": "b4963dda9b31080be1905ef085ecd7dd6cd47c05c79b9cdf83ade83ab2ab271a",
|
||||
"https://bcr.bazel.build/modules/re2/2024-07-02.bcr.1/source.json": "2ff292be6ef3340325ce8a045ecc326e92cbfab47c7cbab4bd85d28971b97ac4",
|
||||
"https://bcr.bazel.build/modules/re2/2024-07-02/MODULE.bazel": "0eadc4395959969297cbcf31a249ff457f2f1d456228c67719480205aa306daa",
|
||||
"https://bcr.bazel.build/modules/re2/2024-07-02/source.json": "547d0111a9d4f362db32196fef805abbf3676e8d6afbe44d395d87816c1130ca",
|
||||
"https://bcr.bazel.build/modules/rules_android/0.1.1/MODULE.bazel": "48809ab0091b07ad0182defb787c4c5328bd3a278938415c00a7b69b50c4d3a8",
|
||||
"https://bcr.bazel.build/modules/rules_android/0.1.1/source.json": "e6986b41626ee10bdc864937ffb6d6bf275bb5b9c65120e6137d56e6331f089e",
|
||||
"https://bcr.bazel.build/modules/rules_apple/3.2.1/MODULE.bazel": "55f19b572fdb4a4bd5a1c2231b60c663f8b4e3146769062b2bdeba72226c74b7",
|
||||
"https://bcr.bazel.build/modules/rules_apple/3.2.1/source.json": "b7e8aa85c5fcd9dfec8ea62642b7e93ca161a03b2c403ae05872813f5dfeb7be",
|
||||
"https://bcr.bazel.build/modules/rules_apple/3.17.0/MODULE.bazel": "39ee3d15edbf09f536a52c42b4297fe1f08a239e28964555f578f301159033f3",
|
||||
"https://bcr.bazel.build/modules/rules_apple/3.17.0/source.json": "57af375068c7292d88062b2a3c2cae546f6e262f653f03388b5d73207f7482e3",
|
||||
"https://bcr.bazel.build/modules/rules_cc/0.0.1/MODULE.bazel": "cb2aa0747f84c6c3a78dad4e2049c154f08ab9d166b1273835a8174940365647",
|
||||
"https://bcr.bazel.build/modules/rules_cc/0.0.10/MODULE.bazel": "ec1705118f7eaedd6e118508d3d26deba2a4e76476ada7e0e3965211be012002",
|
||||
"https://bcr.bazel.build/modules/rules_cc/0.0.13/MODULE.bazel": "0e8529ed7b323dad0775ff924d2ae5af7640b23553dfcd4d34344c7e7a867191",
|
||||
@ -350,10 +347,10 @@
|
||||
"https://bcr.bazel.build/modules/rules_rust/0.57.1/source.json": "deb97fb4b4e7c04adb7d95c21e1b845d5369faa98f3f021c525d20342c3994e0",
|
||||
"https://bcr.bazel.build/modules/rules_shell/0.2.0/MODULE.bazel": "fda8a652ab3c7d8fee214de05e7a9916d8b28082234e8d2c0094505c5268ed3c",
|
||||
"https://bcr.bazel.build/modules/rules_shell/0.3.0/MODULE.bazel": "de4402cd12f4cc8fda2354fce179fdb068c0b9ca1ec2d2b17b3e21b24c1a937b",
|
||||
"https://bcr.bazel.build/modules/rules_shell/0.4.0/MODULE.bazel": "0f8f11bb3cd11755f0b48c1de0bbcf62b4b34421023aa41a2fc74ef68d9584f0",
|
||||
"https://bcr.bazel.build/modules/rules_shell/0.4.0/source.json": "1d7fa7f941cd41dc2704ba5b4edc2e2230eea1cc600d80bd2b65838204c50b95",
|
||||
"https://bcr.bazel.build/modules/rules_swift/1.15.1/MODULE.bazel": "7031d3e5221b54dfee3901d221f10f54a8b0f2a64d4616de7879bcf83fe8d85d",
|
||||
"https://bcr.bazel.build/modules/rules_swift/1.15.1/source.json": "6472a4607d0a2f39eeb7daae78054b995aba593caad99aa64561f631fcf93a63",
|
||||
"https://bcr.bazel.build/modules/rules_shell/0.3.0/source.json": "c55ed591aa5009401ddf80ded9762ac32c358d2517ee7820be981e2de9756cf3",
|
||||
"https://bcr.bazel.build/modules/rules_swift/1.16.0/MODULE.bazel": "4a09f199545a60d09895e8281362b1ff3bb08bbde69c6fc87aff5b92fcc916ca",
|
||||
"https://bcr.bazel.build/modules/rules_swift/2.1.1/MODULE.bazel": "494900a80f944fc7aa61500c2073d9729dff0b764f0e89b824eb746959bc1046",
|
||||
"https://bcr.bazel.build/modules/rules_swift/2.1.1/source.json": "40fc69dfaac64deddbb75bd99cdac55f4427d9ca0afbe408576a65428427a186",
|
||||
"https://bcr.bazel.build/modules/rules_uv/0.53.0/MODULE.bazel": "795c21aaad2574f8cb8163a03808ef4a4a33c0b0a81c0749ca489563561b83c1",
|
||||
"https://bcr.bazel.build/modules/rules_uv/0.53.0/source.json": "bd665e54519407989760844261e02b184ea4f141e65e399f75cd6a3ebf02b649",
|
||||
"https://bcr.bazel.build/modules/stardoc/0.5.1/MODULE.bazel": "1a05d92974d0c122f5ccf09291442580317cdd859f07a8655f1db9a60374f9f8",
|
||||
@ -363,6 +360,8 @@
|
||||
"https://bcr.bazel.build/modules/stardoc/0.7.0/MODULE.bazel": "05e3d6d30c099b6770e97da986c53bd31844d7f13d41412480ea265ac9e8079c",
|
||||
"https://bcr.bazel.build/modules/stardoc/0.7.1/MODULE.bazel": "3548faea4ee5dda5580f9af150e79d0f6aea934fc60c1cc50f4efdd9420759e7",
|
||||
"https://bcr.bazel.build/modules/stardoc/0.7.1/source.json": "b6500ffcd7b48cd72c29bb67bcac781e12701cc0d6d55d266a652583cfcdab01",
|
||||
"https://bcr.bazel.build/modules/swift_argument_parser/1.3.1.1/MODULE.bazel": "5e463fbfba7b1701d957555ed45097d7f984211330106ccd1352c6e0af0dcf91",
|
||||
"https://bcr.bazel.build/modules/swift_argument_parser/1.3.1.1/source.json": "32bd87e5f4d7acc57c5b2ff7c325ae3061d5e242c0c4c214ae87e0f1c13e54cb",
|
||||
"https://bcr.bazel.build/modules/toolchains_protoc/0.3.7/MODULE.bazel": "d2758e5af5838970175274c4fcb67edc8d2f4f8f348885da90c42fafc221236d",
|
||||
"https://bcr.bazel.build/modules/toolchains_protoc/0.3.7/source.json": "07e0eab9be4aded2a8074504a80c55c5a48e811a9824061642af639baaabf985",
|
||||
"https://bcr.bazel.build/modules/upb/0.0.0-20220923-a547704/MODULE.bazel": "7298990c00040a0e2f121f6c32544bab27d4452f80d9ce51349b1a28f3005c43",
|
||||
@ -382,7 +381,7 @@
|
||||
"@@apple_support+//crosstool:setup.bzl%apple_cc_configure_extension": {
|
||||
"general": {
|
||||
"bzlTransitiveDigest": "WJQmiWkUslCPJV2WR2alN3EJBaLsBEr0VYitFeUShw8=",
|
||||
"usagesDigest": "GXdVj/JKgZQLu/JlKFRmqaEA7K3H6OtmWaIYGPAUMGM=",
|
||||
"usagesDigest": "wD+r6kULzwkrPbJnhFpRl2hIsJZIwSSaeEUj3z6v0i4=",
|
||||
"recordedFileInputs": {},
|
||||
"recordedDirentsInputs": {},
|
||||
"envVariables": {},
|
||||
@ -731,10 +730,87 @@
|
||||
]
|
||||
}
|
||||
},
|
||||
"@@rules_apple+//apple:apple.bzl%provisioning_profile_repository_extension": {
|
||||
"general": {
|
||||
"bzlTransitiveDigest": "YmN6XL9GakS6hrKdREw1YK+PQXQDcm/XepqLxIq5s7I=",
|
||||
"usagesDigest": "SiFU9DJJgA6RGhMdCKp9MfHJCxWgKRqtl4gMgcaLyoE=",
|
||||
"recordedFileInputs": {},
|
||||
"recordedDirentsInputs": {},
|
||||
"envVariables": {},
|
||||
"generatedRepoSpecs": {
|
||||
"local_provisioning_profiles": {
|
||||
"repoRuleId": "@@rules_apple+//apple/internal:local_provisioning_profiles.bzl%provisioning_profile_repository",
|
||||
"attributes": {}
|
||||
}
|
||||
},
|
||||
"recordedRepoMappingEntries": [
|
||||
[
|
||||
"apple_support+",
|
||||
"bazel_skylib",
|
||||
"bazel_skylib+"
|
||||
],
|
||||
[
|
||||
"bazel_tools",
|
||||
"rules_cc",
|
||||
"rules_cc+"
|
||||
],
|
||||
[
|
||||
"rules_apple+",
|
||||
"bazel_skylib",
|
||||
"bazel_skylib+"
|
||||
],
|
||||
[
|
||||
"rules_apple+",
|
||||
"bazel_tools",
|
||||
"bazel_tools"
|
||||
],
|
||||
[
|
||||
"rules_apple+",
|
||||
"build_bazel_apple_support",
|
||||
"apple_support+"
|
||||
],
|
||||
[
|
||||
"rules_apple+",
|
||||
"build_bazel_rules_swift",
|
||||
"rules_swift+"
|
||||
],
|
||||
[
|
||||
"rules_cc+",
|
||||
"bazel_tools",
|
||||
"bazel_tools"
|
||||
],
|
||||
[
|
||||
"rules_swift+",
|
||||
"bazel_skylib",
|
||||
"bazel_skylib+"
|
||||
],
|
||||
[
|
||||
"rules_swift+",
|
||||
"bazel_tools",
|
||||
"bazel_tools"
|
||||
],
|
||||
[
|
||||
"rules_swift+",
|
||||
"build_bazel_apple_support",
|
||||
"apple_support+"
|
||||
],
|
||||
[
|
||||
"rules_swift+",
|
||||
"build_bazel_rules_swift",
|
||||
"rules_swift+"
|
||||
],
|
||||
[
|
||||
"rules_swift+",
|
||||
"build_bazel_rules_swift_local_config",
|
||||
"rules_swift++non_module_deps+build_bazel_rules_swift_local_config"
|
||||
]
|
||||
]
|
||||
}
|
||||
},
|
||||
"@@rules_apple+//apple:extensions.bzl%non_module_deps": {
|
||||
"general": {
|
||||
"bzlTransitiveDigest": "oXP4s7ZVB0s16EczTU/VMPd9drP/G5wFRd+WKkF3jX8=",
|
||||
"usagesDigest": "xFNdgn04GvtszsLtPpsiTpPvhsO5ewt/gSh3J2lMZng=",
|
||||
"bzlTransitiveDigest": "asdaC8zi+Dyuf7pGjIXOD+k5Z5KuWzlWe38OXiK9lls=",
|
||||
"usagesDigest": "/KYD+JFOPPK0SVlsHZAG7VLxscWl5XFZW13XI73CUlk=",
|
||||
"recordedFileInputs": {},
|
||||
"recordedDirentsInputs": {},
|
||||
"envVariables": {},
|
||||
@ -968,7 +1044,7 @@
|
||||
},
|
||||
"@@rules_python+//python/private/pypi:pip.bzl%pip_internal": {
|
||||
"general": {
|
||||
"bzlTransitiveDigest": "Uk/Dp12Z3OcPmLAsj/LI7OiixfCIAQYEAPq01vXZHTY=",
|
||||
"bzlTransitiveDigest": "WQT5YLJiIaCB8kyx35MGdMpQFtljg0stNXsxxbcXvzA=",
|
||||
"usagesDigest": "OLoIStnzNObNalKEMRq99FqenhPGLFZ5utVLV4sz7OI=",
|
||||
"recordedFileInputs": {
|
||||
"@@rules_python+//tools/publish/requirements_darwin.txt": "2c51d7903dc21cbe23d9b35ce6592f463343aed8a166de445020024539b627fc",
|
||||
@ -3462,12 +3538,19 @@
|
||||
},
|
||||
"@@rules_rust+//cargo/private:internal_extensions.bzl%i": {
|
||||
"general": {
|
||||
"bzlTransitiveDigest": "HzxsZGj/ooGpRGyFWvbEbjKjVQ0bRsqNFC16HgiBoLA=",
|
||||
"usagesDigest": "ghwjHh0kIp++0LV3XUhmk1+gm8qw9h7zVwXaLXeylvg=",
|
||||
"bzlTransitiveDigest": "KHxps6vBZ0grkMXc4wUDyxHy4g1RX5nSm+SHIjDUwFU=",
|
||||
"usagesDigest": "44NapJNcDDl7g/avr9Z+huiyJvLuC+hsktRy/1rZBIk=",
|
||||
"recordedFileInputs": {},
|
||||
"recordedDirentsInputs": {},
|
||||
"envVariables": {},
|
||||
"generatedRepoSpecs": {
|
||||
"rrcti": {
|
||||
"repoRuleId": "@@rules_rust+//crate_universe/private:crates_vendor.bzl%crates_vendor_remote_repository",
|
||||
"attributes": {
|
||||
"build_file": "@@rules_rust+//cargo/private/cargo_toml_info/3rdparty/crates:BUILD.bazel",
|
||||
"defs_module": "@@rules_rust+//cargo/private/cargo_toml_info/3rdparty/crates:defs.bzl"
|
||||
}
|
||||
},
|
||||
"rrcti__cargo_toml-0.20.5": {
|
||||
"repoRuleId": "@@bazel_tools//tools/build_defs/repo:http.bzl%http_archive",
|
||||
"attributes": {
|
||||
@ -3663,6 +3746,7 @@
|
||||
},
|
||||
"moduleExtensionMetadata": {
|
||||
"explicitRootModuleDirectDeps": [
|
||||
"rrcti",
|
||||
"rrcti__cargo_toml-0.20.5"
|
||||
],
|
||||
"explicitRootModuleDirectDevDeps": [],
|
||||
@ -3683,14 +3767,19 @@
|
||||
[
|
||||
"rules_rust+",
|
||||
"rrcti",
|
||||
"rrcti"
|
||||
"rules_rust++i2+rrcti"
|
||||
],
|
||||
[
|
||||
"rules_rust+",
|
||||
"rules_rust",
|
||||
"rules_rust+"
|
||||
]
|
||||
]
|
||||
}
|
||||
},
|
||||
"@@rules_rust+//crate_universe:extensions.bzl%crate": {
|
||||
"general": {
|
||||
"bzlTransitiveDigest": "AZhzD4bwYRydNmUBNtxFFnszLhskORJZ0JIEp+ES0Tw=",
|
||||
"bzlTransitiveDigest": "wCEZ3jfLKpigLStmA+7HYbnchHAogZXm6LI/ugF/gfw=",
|
||||
"usagesDigest": "LRFoDTnUPiFo9Sf1CJ7R6NZDugjywFLeLDfqQzz9zEA=",
|
||||
"recordedFileInputs": {
|
||||
"@@//zml/tokenizer/hftokenizers/Cargo.toml": "b0b28872bb74e13aec7ef0ef8dfd5ec99a8573bc12085c0f36711f2a87912061"
|
||||
@ -3701,7 +3790,7 @@
|
||||
"CARGO_BAZEL_GENERATOR_SHA256": null,
|
||||
"CARGO_BAZEL_GENERATOR_URL": null,
|
||||
"CARGO_BAZEL_ISOLATED": null,
|
||||
"CARGO_BAZEL_REPIN": "1",
|
||||
"CARGO_BAZEL_REPIN": null,
|
||||
"CARGO_BAZEL_REPIN_ONLY": null,
|
||||
"REPIN": null
|
||||
},
|
||||
@ -4940,8 +5029,8 @@
|
||||
},
|
||||
"@@rules_rust+//crate_universe/private:internal_extensions.bzl%cu_nr": {
|
||||
"general": {
|
||||
"bzlTransitiveDigest": "IiqHICNdoz5IUOyuGtBHtVW5aqS9OjFQktWJTM4voSc=",
|
||||
"usagesDigest": "y6BFi/0Udc5/Shj1dxUxX4jfu5ybjfOiQhztUDSA7ws=",
|
||||
"bzlTransitiveDigest": "Ro7qPeo3MPsw94F7OZRkQYnLg140aO3CFF4EqSee28s=",
|
||||
"usagesDigest": "JTRjQHmqGCK2FUQYDLQD4VejKtuKKC8LrOQhFOpHPPo=",
|
||||
"recordedFileInputs": {},
|
||||
"recordedDirentsInputs": {},
|
||||
"envVariables": {},
|
||||
@ -5078,8 +5167,8 @@
|
||||
},
|
||||
"@@rules_swift+//swift:extensions.bzl%non_module_deps": {
|
||||
"general": {
|
||||
"bzlTransitiveDigest": "urnhfq3NjI9wFj7Q1nbFjfGu21BGKMdrc7P8v+mEzyA=",
|
||||
"usagesDigest": "oxicQmmPKp7Hz7aNyhnahbwtAN0bJqP+huWfVh1AngQ=",
|
||||
"bzlTransitiveDigest": "6U1cwn6fF8hXqLn4cawhmQ3814PCS4XbSG532q0X9qI=",
|
||||
"usagesDigest": "mhACFnrdMv9Wi0Mt67bxocJqviRkDSV+Ee5Mqdj5akA=",
|
||||
"recordedFileInputs": {},
|
||||
"recordedDirentsInputs": {},
|
||||
"envVariables": {},
|
||||
@ -5106,6 +5195,17 @@
|
||||
"build_file": "@@rules_swift+//third_party:com_github_grpc_grpc_swift/BUILD.overlay"
|
||||
}
|
||||
},
|
||||
"com_github_apple_swift_docc_symbolkit": {
|
||||
"repoRuleId": "@@bazel_tools//tools/build_defs/repo:http.bzl%http_archive",
|
||||
"attributes": {
|
||||
"urls": [
|
||||
"https://github.com/apple/swift-docc-symbolkit/archive/refs/tags/swift-5.10-RELEASE.tar.gz"
|
||||
],
|
||||
"sha256": "de1d4b6940468ddb53b89df7aa1a81323b9712775b0e33e8254fa0f6f7469a97",
|
||||
"strip_prefix": "swift-docc-symbolkit-swift-5.10-RELEASE",
|
||||
"build_file": "@@rules_swift+//third_party:com_github_apple_swift_docc_symbolkit/BUILD.overlay"
|
||||
}
|
||||
},
|
||||
"com_github_apple_swift_nio": {
|
||||
"repoRuleId": "@@bazel_tools//tools/build_defs/repo:http.bzl%http_archive",
|
||||
"attributes": {
|
||||
|
||||
@ -22,15 +22,15 @@ def _cpu_pjrt_plugin_impl(mctx):
|
||||
http_archive(
|
||||
name = "libpjrt_cpu_linux_amd64",
|
||||
build_file_content = _BUILD_LINUX,
|
||||
sha256 = "0f2cb204015e062df5d1cbd39d8c01c076ab2b004d0f4f37f6d5e120d3cd7087",
|
||||
url = "https://github.com/zml/pjrt-artifacts/releases/download/v5.0.0/pjrt-cpu_linux-amd64.tar.gz",
|
||||
sha256 = "e17d15331a3f42b90c8131459c235a9cf1145913a581c8b192845c50d313a7d6",
|
||||
url = "https://github.com/zml/pjrt-artifacts/releases/download/v6.0.0/pjrt-cpu_linux-amd64.tar.gz",
|
||||
)
|
||||
|
||||
http_archive(
|
||||
name = "libpjrt_cpu_darwin_arm64",
|
||||
build_file_content = _BUILD_DARWIN,
|
||||
sha256 = "2ddb66a93c8a913e3bc8f291e01df59aa297592cc91e05aab2dd4813884098cb",
|
||||
url = "https://github.com/zml/pjrt-artifacts/releases/download/v5.0.0/pjrt-cpu_darwin-arm64.tar.gz",
|
||||
sha256 = "2e18f8426ec5fa698163d55871fdbaed3616325cab10fad54f9d14f265fbf00d",
|
||||
url = "https://github.com/zml/pjrt-artifacts/releases/download/v6.0.0/pjrt-cpu_darwin-arm64.tar.gz",
|
||||
)
|
||||
|
||||
return mctx.extension_metadata(
|
||||
|
||||
@ -208,9 +208,8 @@ def _cuda_impl(mctx):
|
||||
http_archive(
|
||||
name = "libpjrt_cuda",
|
||||
build_file = "libpjrt_cuda.BUILD.bazel",
|
||||
url = "https://files.pythonhosted.org/packages/90/43/ac2c369e202e3e3e7e5aa7929b197801ba02eaf11868437adaa5341704e4/jax_cuda12_pjrt-0.4.38-py3-none-manylinux2014_x86_64.whl",
|
||||
type = "zip",
|
||||
sha256 = "83be4c59fbcf30077a60085d98e7d59dc738b1c91e0d628e4ac1779fde15ac2b",
|
||||
url = "https://github.com/zml/pjrt-artifacts/releases/download/v6.0.0/pjrt-cuda_linux-amd64.tar.gz",
|
||||
sha256 = "7ee04fdacee1d144bc14a47ab0c453e600f6a821dea2ecb1d5af9b7093038dbf",
|
||||
)
|
||||
|
||||
return mctx.extension_metadata(
|
||||
|
||||
@ -25,8 +25,7 @@ copy_to_directory(
|
||||
cc_import(
|
||||
name = "libpjrt_cuda",
|
||||
data = [":sandbox"],
|
||||
shared_library = "jax_plugins/xla_cuda12/xla_cuda_plugin.so",
|
||||
soname = "libpjrt_cuda.so",
|
||||
shared_library = "libpjrt_cuda.so",
|
||||
add_needed = ["libzmlxcuda.so.0"],
|
||||
rename_dynamic_symbols = {
|
||||
"dlopen": "zmlxcuda_dlopen",
|
||||
|
||||
@ -212,13 +212,13 @@ def _rocm_impl(mctx):
|
||||
http_archive(
|
||||
name = "libpjrt_rocm",
|
||||
build_file = "libpjrt_rocm.BUILD.bazel",
|
||||
url = "https://github.com/zml/pjrt-artifacts/releases/download/v5.0.0/pjrt-rocm_linux-amd64.tar.gz",
|
||||
sha256 = "2c7a687827f63987caa117cd5b56a6e20291681ae1c51edd54241a1181e91d2d",
|
||||
url = "https://github.com/zml/pjrt-artifacts/releases/download/v6.0.0/pjrt-rocm_linux-amd64.tar.gz",
|
||||
sha256 = "0a41e5b255d0b3284e029c11c2362d777557f87778e1531a2117bc46b835395a",
|
||||
)
|
||||
|
||||
return mctx.extension_metadata(
|
||||
reproducible = True,
|
||||
root_module_direct_deps = ["libpjrt_rocm"],
|
||||
root_module_direct_deps = ["libpjrt_rocm", "hipblaslt", "rocblas"],
|
||||
root_module_direct_dev_deps = [],
|
||||
)
|
||||
|
||||
|
||||
@ -48,7 +48,7 @@ fn setupRocmEnv() !void {
|
||||
for (rocm_env_entries) |entry| {
|
||||
var real_path = r.rlocationAlloc(arena.allocator(), entry.rpath) catch null orelse {
|
||||
if (entry.mandatory) {
|
||||
stdx.debug.panic("Unable to find {s} in {s}", .{ entry.name, bazel_builtin.current_repository });
|
||||
stdx.debug.panic("Unable to find {s} in {s}\n", .{ entry.name, bazel_builtin.current_repository });
|
||||
}
|
||||
continue;
|
||||
};
|
||||
|
||||
11
third_party/modules/llvm-raw/20250201.0-386af4a/MODULE.bazel
vendored
Normal file
11
third_party/modules/llvm-raw/20250201.0-386af4a/MODULE.bazel
vendored
Normal file
@ -0,0 +1,11 @@
|
||||
module(
|
||||
name = "llvm-raw",
|
||||
version = "20250201.0-386af4a",
|
||||
compatibility_level = 1,
|
||||
)
|
||||
|
||||
bazel_dep(name = "bazel_skylib", version = "1.7.1")
|
||||
bazel_dep(name = "platforms", version = "0.0.11")
|
||||
bazel_dep(name = "zstd", version = "1.5.6", repo_name = "llvm_zstd")
|
||||
bazel_dep(name = "zlib", version = "1.3.1.bcr.3", repo_name = "llvm_zlib")
|
||||
bazel_dep(name = "rules_python", version = "0.29.0")
|
||||
0
third_party/modules/llvm-raw/20250201.0-386af4a/overlay/BUILD.bazel
vendored
Normal file
0
third_party/modules/llvm-raw/20250201.0-386af4a/overlay/BUILD.bazel
vendored
Normal file
11
third_party/modules/llvm-raw/20250201.0-386af4a/overlay/MODULE.bazel
vendored
Normal file
11
third_party/modules/llvm-raw/20250201.0-386af4a/overlay/MODULE.bazel
vendored
Normal file
@ -0,0 +1,11 @@
|
||||
module(
|
||||
name = "llvm-raw",
|
||||
version = "20250201.0-386af4a",
|
||||
compatibility_level = 1,
|
||||
)
|
||||
|
||||
bazel_dep(name = "bazel_skylib", version = "1.7.1")
|
||||
bazel_dep(name = "platforms", version = "0.0.11")
|
||||
bazel_dep(name = "zstd", version = "1.5.6", repo_name = "llvm_zstd")
|
||||
bazel_dep(name = "zlib", version = "1.3.1.bcr.3", repo_name = "llvm_zlib")
|
||||
bazel_dep(name = "rules_python", version = "0.29.0")
|
||||
28
third_party/modules/llvm-raw/20250201.0-386af4a/overlay/utils/bazel/extension.bzl
vendored
Normal file
28
third_party/modules/llvm-raw/20250201.0-386af4a/overlay/utils/bazel/extension.bzl
vendored
Normal file
@ -0,0 +1,28 @@
|
||||
load("//utils/bazel:configure.bzl", _llvm_configure = "llvm_configure")
|
||||
|
||||
def _llvm_impl(mctx):
|
||||
_targets = {}
|
||||
for mod in mctx.modules:
|
||||
for conf in mod.tags.configure:
|
||||
for target in conf.targets:
|
||||
_targets[target] = True
|
||||
_llvm_configure(
|
||||
name = "llvm-project",
|
||||
targets = _targets.keys(),
|
||||
)
|
||||
return mctx.extension_metadata(
|
||||
reproducible = True,
|
||||
root_module_direct_deps = "all",
|
||||
root_module_direct_dev_deps = [],
|
||||
)
|
||||
|
||||
llvm = module_extension(
|
||||
implementation = _llvm_impl,
|
||||
tag_classes = {
|
||||
"configure": tag_class(
|
||||
attrs = {
|
||||
"targets": attr.string_list(mandatory = True),
|
||||
},
|
||||
),
|
||||
},
|
||||
)
|
||||
10
third_party/modules/llvm-raw/20250201.0-386af4a/source.json
vendored
Normal file
10
third_party/modules/llvm-raw/20250201.0-386af4a/source.json
vendored
Normal file
@ -0,0 +1,10 @@
|
||||
{
|
||||
"strip_prefix": "llvm-project-386af4a5c64ab75eaee2448dc38f2e34a40bfed0",
|
||||
"url": "https://github.com/llvm/llvm-project/archive/386af4a5c64ab75eaee2448dc38f2e34a40bfed0.tar.gz",
|
||||
"integrity": "sha256-S3UqLfteTlXPmBKjU7zn5DfghS9O72SSn0jXRzIIDFA=",
|
||||
"overlay": {
|
||||
"BUILD.bazel": "",
|
||||
"MODULE.bazel": "",
|
||||
"utils/bazel/extension.bzl": ""
|
||||
}
|
||||
}
|
||||
1
third_party/modules/llvm-raw/metadata.json
vendored
1
third_party/modules/llvm-raw/metadata.json
vendored
@ -16,6 +16,7 @@
|
||||
"20241022.0-6c4267f",
|
||||
"20250102.0-f739aa4",
|
||||
"20250117.0-bf17016",
|
||||
"20250201.0-386af4a",
|
||||
],
|
||||
"yanked_versions": {}
|
||||
}
|
||||
|
||||
15
third_party/modules/stablehlo/20250201.0-7775e3e/MODULE.bazel
vendored
Normal file
15
third_party/modules/stablehlo/20250201.0-7775e3e/MODULE.bazel
vendored
Normal file
@ -0,0 +1,15 @@
|
||||
module(
|
||||
name = "stablehlo",
|
||||
version = "20250201.0-7775e3e",
|
||||
compatibility_level = 1,
|
||||
)
|
||||
|
||||
bazel_dep(name = "bazel_skylib", version = "1.7.1")
|
||||
bazel_dep(name = "rules_cc", version = "0.0.9")
|
||||
bazel_dep(name = "llvm-raw", version = "20250201.0-386af4a")
|
||||
|
||||
llvm = use_extension("@llvm-raw//utils/bazel:extension.bzl", "llvm")
|
||||
llvm.configure(
|
||||
targets = ["AArch64", "X86", "NVPTX"],
|
||||
)
|
||||
use_repo(llvm, "llvm-project")
|
||||
15
third_party/modules/stablehlo/20250201.0-7775e3e/overlay/MODULE.bazel
vendored
Normal file
15
third_party/modules/stablehlo/20250201.0-7775e3e/overlay/MODULE.bazel
vendored
Normal file
@ -0,0 +1,15 @@
|
||||
module(
|
||||
name = "stablehlo",
|
||||
version = "20250201.0-7775e3e",
|
||||
compatibility_level = 1,
|
||||
)
|
||||
|
||||
bazel_dep(name = "bazel_skylib", version = "1.7.1")
|
||||
bazel_dep(name = "rules_cc", version = "0.0.9")
|
||||
bazel_dep(name = "llvm-raw", version = "20250201.0-386af4a")
|
||||
|
||||
llvm = use_extension("@llvm-raw//utils/bazel:extension.bzl", "llvm")
|
||||
llvm.configure(
|
||||
targets = ["AArch64", "X86", "NVPTX"],
|
||||
)
|
||||
use_repo(llvm, "llvm-project")
|
||||
8
third_party/modules/stablehlo/20250201.0-7775e3e/source.json
vendored
Normal file
8
third_party/modules/stablehlo/20250201.0-7775e3e/source.json
vendored
Normal file
@ -0,0 +1,8 @@
|
||||
{
|
||||
"strip_prefix": "stablehlo-7775e3e2f98fe458f227cdc41aec567535d1c796",
|
||||
"url": "https://github.com/openxla/stablehlo/archive/7775e3e2f98fe458f227cdc41aec567535d1c796.tar.gz",
|
||||
"integrity": "sha256-HM/MIs3FeN4PxcYgFsjj04tVJ4u4usGiYGSa9BwT5bg=",
|
||||
"overlay": {
|
||||
"MODULE.bazel": ""
|
||||
}
|
||||
}
|
||||
1
third_party/modules/stablehlo/metadata.json
vendored
1
third_party/modules/stablehlo/metadata.json
vendored
@ -16,6 +16,7 @@
|
||||
"20241021.0-1c0b606",
|
||||
"20241220.0-38bb2f9",
|
||||
"20250117.0-c125b32",
|
||||
"20250201.0-7775e3e",
|
||||
],
|
||||
"yanked_versions": {}
|
||||
}
|
||||
|
||||
34
third_party/modules/xla/20250204-0-6789523/MODULE.bazel
vendored
Normal file
34
third_party/modules/xla/20250204-0-6789523/MODULE.bazel
vendored
Normal file
@ -0,0 +1,34 @@
|
||||
module(
|
||||
name = "xla",
|
||||
version = "20250204-0-6789523",
|
||||
compatibility_level = 1,
|
||||
)
|
||||
|
||||
bazel_dep(name = "platforms", version = "0.0.8")
|
||||
bazel_dep(name = "bazel_skylib", version = "1.5.0")
|
||||
bazel_dep(name = "rules_cc", version = "0.0.17")
|
||||
bazel_dep(name = "rules_apple", version = "3.17.0", repo_name = "build_bazel_rules_apple")
|
||||
bazel_dep(name = "abseil-cpp", version = "20240116.0", repo_name = "com_google_absl")
|
||||
bazel_dep(name = "rules_python", version = "0.29.0")
|
||||
bazel_dep(name = "rules_proto", version = "6.0.0-rc1")
|
||||
bazel_dep(name = "rules_java", version = "7.3.2")
|
||||
bazel_dep(name = "rules_pkg", version = "0.9.1")
|
||||
bazel_dep(name = "zlib", version = "1.2.13")
|
||||
bazel_dep(name = "re2", version = "2024-07-02.bcr.1", repo_name = "com_googlesource_code_re2")
|
||||
bazel_dep(name = "rules_license", version = "0.0.8")
|
||||
|
||||
bazel_dep(name = "stablehlo", version = "20250201.0-7775e3e")
|
||||
|
||||
tsl = use_extension("//:tsl.bzl", "tsl")
|
||||
use_repo(tsl, "tsl")
|
||||
|
||||
xla_workspace = use_extension("//:workspace.bzl", "xla_workspace")
|
||||
use_repo(
|
||||
xla_workspace,
|
||||
"com_github_grpc_grpc",
|
||||
"com_google_protobuf",
|
||||
"local_config_cuda",
|
||||
"local_config_remote_execution",
|
||||
"local_config_rocm",
|
||||
"local_config_tensorrt",
|
||||
)
|
||||
34
third_party/modules/xla/20250204-0-6789523/overlay/MODULE.bazel
vendored
Normal file
34
third_party/modules/xla/20250204-0-6789523/overlay/MODULE.bazel
vendored
Normal file
@ -0,0 +1,34 @@
|
||||
module(
|
||||
name = "xla",
|
||||
version = "20250204-0-6789523",
|
||||
compatibility_level = 1,
|
||||
)
|
||||
|
||||
bazel_dep(name = "platforms", version = "0.0.8")
|
||||
bazel_dep(name = "bazel_skylib", version = "1.5.0")
|
||||
bazel_dep(name = "rules_cc", version = "0.0.17")
|
||||
bazel_dep(name = "rules_apple", version = "3.17.0", repo_name = "build_bazel_rules_apple")
|
||||
bazel_dep(name = "abseil-cpp", version = "20240116.0", repo_name = "com_google_absl")
|
||||
bazel_dep(name = "rules_python", version = "0.29.0")
|
||||
bazel_dep(name = "rules_proto", version = "6.0.0-rc1")
|
||||
bazel_dep(name = "rules_java", version = "7.3.2")
|
||||
bazel_dep(name = "rules_pkg", version = "0.9.1")
|
||||
bazel_dep(name = "zlib", version = "1.2.13")
|
||||
bazel_dep(name = "re2", version = "2024-07-02.bcr.1", repo_name = "com_googlesource_code_re2")
|
||||
bazel_dep(name = "rules_license", version = "0.0.8")
|
||||
|
||||
bazel_dep(name = "stablehlo", version = "20250201.0-7775e3e")
|
||||
|
||||
tsl = use_extension("//:tsl.bzl", "tsl")
|
||||
use_repo(tsl, "tsl")
|
||||
|
||||
xla_workspace = use_extension("//:workspace.bzl", "xla_workspace")
|
||||
use_repo(
|
||||
xla_workspace,
|
||||
"com_github_grpc_grpc",
|
||||
"com_google_protobuf",
|
||||
"local_config_cuda",
|
||||
"local_config_remote_execution",
|
||||
"local_config_rocm",
|
||||
"local_config_tensorrt",
|
||||
)
|
||||
19
third_party/modules/xla/20250204-0-6789523/overlay/tsl.bzl
vendored
Normal file
19
third_party/modules/xla/20250204-0-6789523/overlay/tsl.bzl
vendored
Normal file
@ -0,0 +1,19 @@
|
||||
load("//third_party:repo.bzl", "tf_vendored")
|
||||
load("//third_party/py:python_init_repositories.bzl", "python_init_repositories")
|
||||
|
||||
def _tsl_impl(mctx):
|
||||
python_init_repositories(
|
||||
requirements = {
|
||||
"3.11": "//:requirements_lock_3_11.txt",
|
||||
},
|
||||
)
|
||||
tf_vendored(name = "tsl", relpath = "third_party/tsl")
|
||||
return mctx.extension_metadata(
|
||||
reproducible = True,
|
||||
root_module_direct_deps = ["tsl"],
|
||||
root_module_direct_dev_deps = [],
|
||||
)
|
||||
|
||||
tsl = module_extension(
|
||||
implementation = _tsl_impl,
|
||||
)
|
||||
52
third_party/modules/xla/20250204-0-6789523/overlay/workspace.bzl
vendored
Normal file
52
third_party/modules/xla/20250204-0-6789523/overlay/workspace.bzl
vendored
Normal file
@ -0,0 +1,52 @@
|
||||
load("@tsl//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
|
||||
load("@tsl//third_party/gpus:cuda_configure.bzl", "cuda_configure")
|
||||
load("@tsl//third_party/gpus:rocm_configure.bzl", "rocm_configure")
|
||||
load("@tsl//third_party/tensorrt:tensorrt_configure.bzl", "tensorrt_configure")
|
||||
load("@tsl//tools/toolchains/remote:configure.bzl", "remote_execution_configure")
|
||||
|
||||
def _xla_workspace_impl(mctx):
|
||||
cuda_configure(name = "local_config_cuda")
|
||||
remote_execution_configure(name = "local_config_remote_execution")
|
||||
rocm_configure(name = "local_config_rocm")
|
||||
tensorrt_configure(name = "local_config_tensorrt")
|
||||
tf_http_archive(
|
||||
name = "com_github_grpc_grpc",
|
||||
sha256 = "b956598d8cbe168b5ee717b5dafa56563eb5201a947856a6688bbeac9cac4e1f",
|
||||
strip_prefix = "grpc-b54a5b338637f92bfcf4b0bc05e0f57a5fd8fadd",
|
||||
system_build_file = "@tsl//third_party/systemlibs:grpc.BUILD",
|
||||
patch_file = [
|
||||
"@tsl//third_party/grpc:generate_cc_env_fix.patch",
|
||||
"@tsl//third_party/grpc:register_go_toolchain.patch",
|
||||
],
|
||||
system_link_files = {
|
||||
"@tsl//third_party/systemlibs:BUILD": "bazel/BUILD",
|
||||
"@tsl//third_party/systemlibs:grpc.BUILD": "src/compiler/BUILD",
|
||||
"@tsl//third_party/systemlibs:grpc.bazel.grpc_deps.bzl": "bazel/grpc_deps.bzl",
|
||||
"@tsl//third_party/systemlibs:grpc.bazel.grpc_extra_deps.bzl": "bazel/grpc_extra_deps.bzl",
|
||||
"@tsl//third_party/systemlibs:grpc.bazel.cc_grpc_library.bzl": "bazel/cc_grpc_library.bzl",
|
||||
"@tsl//third_party/systemlibs:grpc.bazel.generate_cc.bzl": "bazel/generate_cc.bzl",
|
||||
"@tsl//third_party/systemlibs:grpc.bazel.protobuf.bzl": "bazel/protobuf.bzl",
|
||||
},
|
||||
urls = tf_mirror_urls("https://github.com/grpc/grpc/archive/b54a5b338637f92bfcf4b0bc05e0f57a5fd8fadd.tar.gz"),
|
||||
)
|
||||
tf_http_archive(
|
||||
name = "com_google_protobuf",
|
||||
patch_file = ["@tsl//third_party/protobuf:protobuf.patch"],
|
||||
sha256 = "f66073dee0bc159157b0bd7f502d7d1ee0bc76b3c1eac9836927511bdc4b3fc1",
|
||||
strip_prefix = "protobuf-3.21.9",
|
||||
system_build_file = "@tsl//third_party/systemlibs:protobuf.BUILD",
|
||||
system_link_files = {
|
||||
"@tsl//third_party/systemlibs:protobuf.bzl": "protobuf.bzl",
|
||||
"@tsl//third_party/systemlibs:protobuf_deps.bzl": "protobuf_deps.bzl",
|
||||
},
|
||||
urls = tf_mirror_urls("https://github.com/protocolbuffers/protobuf/archive/v3.21.9.zip"),
|
||||
)
|
||||
return mctx.extension_metadata(
|
||||
reproducible = True,
|
||||
root_module_direct_deps = "all",
|
||||
root_module_direct_dev_deps = [],
|
||||
)
|
||||
|
||||
xla_workspace = module_extension(
|
||||
implementation = _xla_workspace_impl,
|
||||
)
|
||||
41
third_party/modules/xla/20250204-0-6789523/patches/0001-bazel-migration-to-bazel-8.0.1.patch
vendored
Normal file
41
third_party/modules/xla/20250204-0-6789523/patches/0001-bazel-migration-to-bazel-8.0.1.patch
vendored
Normal file
@ -0,0 +1,41 @@
|
||||
From 6cf475b500521c1b8be06f590fdbc1818f0dc44b Mon Sep 17 00:00:00 2001
|
||||
From: Jean-Baptiste Dalido <jb@zml.ai>
|
||||
Date: Mon, 6 Jan 2025 13:33:13 +0100
|
||||
Subject: [PATCH] bazel: migration to bazel 8.0.1
|
||||
|
||||
---
|
||||
.bazelversion | 2 +-
|
||||
third_party/tsl/third_party/gpus/cuda_configure.bzl | 4 ++--
|
||||
2 files changed, 3 insertions(+), 3 deletions(-)
|
||||
|
||||
diff --git a/.bazelversion b/.bazelversion
|
||||
index f22d756da3..fa5fce04b3 100644
|
||||
--- a/.bazelversion
|
||||
+++ b/.bazelversion
|
||||
@@ -1 +1 @@
|
||||
-6.5.0
|
||||
+8.0.1
|
||||
\ No newline at end of file
|
||||
diff --git a/third_party/tsl/third_party/gpus/cuda_configure.bzl b/third_party/tsl/third_party/gpus/cuda_configure.bzl
|
||||
index d62531152d..71d80a5a99 100644
|
||||
--- a/third_party/tsl/third_party/gpus/cuda_configure.bzl
|
||||
+++ b/third_party/tsl/third_party/gpus/cuda_configure.bzl
|
||||
@@ -33,14 +33,14 @@ NB: DEPRECATED! Use `hermetic/cuda_configure` rule instead.
|
||||
load(
|
||||
"@bazel_tools//tools/cpp:lib_cc_configure.bzl",
|
||||
"escape_string",
|
||||
- "get_env_var",
|
||||
)
|
||||
load(
|
||||
"@bazel_tools//tools/cpp:windows_cc_configure.bzl",
|
||||
- "find_msvc_tool",
|
||||
"find_vc_path",
|
||||
"setup_vc_env_vars",
|
||||
)
|
||||
+load("@rules_cc//cc/private/toolchain:windows_cc_configure.bzl", "find_msvc_tool")
|
||||
+load("@rules_cc//cc/private/toolchain:lib_cc_configure.bzl", "get_env_var")
|
||||
load("//third_party/clang_toolchain:download_clang.bzl", "download_clang")
|
||||
load(
|
||||
"//third_party/remote_config:common.bzl",
|
||||
--
|
||||
2.39.3 (Apple Git-146)
|
||||
@ -0,0 +1,26 @@
|
||||
From 4db5de34f70d991fedbe28915c8239b97ba7a064 Mon Sep 17 00:00:00 2001
|
||||
From: Steeve Morin <steeve.morin@gmail.com>
|
||||
Date: Mon, 18 Mar 2024 17:17:34 +0100
|
||||
Subject: [PATCH 3/3] [PJRT C API] Ensure C compliance for Profiler Extension
|
||||
|
||||
---
|
||||
xla/pjrt/c/pjrt_c_api_profiler_extension.h | 2 ++
|
||||
1 file changed, 2 insertions(+)
|
||||
|
||||
diff --git a/xla/pjrt/c/pjrt_c_api_profiler_extension.h b/xla/pjrt/c/pjrt_c_api_profiler_extension.h
|
||||
index c821916ad..89a596123 100644
|
||||
--- a/xla/pjrt/c/pjrt_c_api_profiler_extension.h
|
||||
+++ b/xla/pjrt/c/pjrt_c_api_profiler_extension.h
|
||||
@@ -16,8 +16,10 @@ limitations under the License.
|
||||
#ifndef XLA_PJRT_C_PJRT_C_API_PROFILER_EXTENSION_H_
|
||||
#define XLA_PJRT_C_PJRT_C_API_PROFILER_EXTENSION_H_
|
||||
|
||||
+#ifdef __cplusplus
|
||||
#include <cstddef>
|
||||
#include <cstdint>
|
||||
+#endif
|
||||
|
||||
#include "xla/backends/profiler/plugin/profiler_c_api.h"
|
||||
#include "xla/pjrt/c/pjrt_c_api.h"
|
||||
--
|
||||
2.39.3 (Apple Git-146)
|
||||
15
third_party/modules/xla/20250204-0-6789523/source.json
vendored
Normal file
15
third_party/modules/xla/20250204-0-6789523/source.json
vendored
Normal file
@ -0,0 +1,15 @@
|
||||
{
|
||||
"strip_prefix": "xla-6789523d0a42c2cf83309a263f417122c4c63811",
|
||||
"url": "https://github.com/openxla/xla/archive/6789523d0a42c2cf83309a263f417122c4c63811.tar.gz",
|
||||
"integrity": "sha256-7z0MdXQBoGApSYpTvYjKFmg18tGkjJ3P/PPQ487cPc8=",
|
||||
"overlay": {
|
||||
"tsl.bzl": "",
|
||||
"workspace.bzl": "",
|
||||
"MODULE.bazel": ""
|
||||
},
|
||||
"patch_strip": 1,
|
||||
"patches": {
|
||||
"0001-bazel-migration-to-bazel-8.0.1.patch": "",
|
||||
"0003-PJRT-C-API-Ensure-C-compliance-for-Profiler-Extensio.patch": ""
|
||||
}
|
||||
}
|
||||
1
third_party/modules/xla/metadata.json
vendored
1
third_party/modules/xla/metadata.json
vendored
@ -17,6 +17,7 @@
|
||||
"20250103.0-5f1fe6a",
|
||||
"20250122.0-cc075be",
|
||||
"20250123.0-cc075be",
|
||||
"20250204-0-6789523",
|
||||
],
|
||||
"yanked_versions": {}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user