tarry.singh/Radix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enable -fllvm flag for CUDA runtime, upb, and ZML Bazel targets.

Browse Source
This commit is contained in:
Tarry Singh 2025-11-11 13:02:40 +00:00
parent 8f50d19cba
commit 3d3a0ea463
5 changed files with 62 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

91
MODULE.bazel.lock
View File

@ -171,6 +171,7 @@
"https://bazel-registry.zml.ai/modules/rules_shell/0.4.1/MODULE.bazel": "not found", "https://bazel-registry.zml.ai/modules/rules_shell/0.4.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/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_swift/2.1.1/MODULE.bazel": "not found",
"https://bazel-registry.zml.ai/modules/rules_zig/0.12.1/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.1/MODULE.bazel": "not found",
"https://bazel-registry.zml.ai/modules/stardoc/0.5.3/MODULE.bazel": "not found", "https://bazel-registry.zml.ai/modules/stardoc/0.5.3/MODULE.bazel": "not found",
"https://bazel-registry.zml.ai/modules/stardoc/0.5.4/MODULE.bazel": "not found", "https://bazel-registry.zml.ai/modules/stardoc/0.5.4/MODULE.bazel": "not found",
@ -398,6 +399,8 @@
"https://bcr.bazel.build/modules/rules_swift/1.16.0/MODULE.bazel": "4a09f199545a60d09895e8281362b1ff3bb08bbde69c6fc87aff5b92fcc916ca", "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/MODULE.bazel": "494900a80f944fc7aa61500c2073d9729dff0b764f0e89b824eb746959bc1046",
"https://bcr.bazel.build/modules/rules_swift/2.1.1/source.json": "40fc69dfaac64deddbb75bd99cdac55f4427d9ca0afbe408576a65428427a186", "https://bcr.bazel.build/modules/rules_swift/2.1.1/source.json": "40fc69dfaac64deddbb75bd99cdac55f4427d9ca0afbe408576a65428427a186",
"https://bcr.bazel.build/modules/rules_zig/0.12.1/MODULE.bazel": "5953094b681c212eebe3dd4275809c52b9177aeaa058d8f1bff433822c0dcd01",
"https://bcr.bazel.build/modules/rules_zig/0.12.1/source.json": "d880cdc3473bfcf172c9aad9292e856430496e198d805539bb8be98e79f58a7c",
"https://bcr.bazel.build/modules/stardoc/0.5.1/MODULE.bazel": "1a05d92974d0c122f5ccf09291442580317cdd859f07a8655f1db9a60374f9f8", "https://bcr.bazel.build/modules/stardoc/0.5.1/MODULE.bazel": "1a05d92974d0c122f5ccf09291442580317cdd859f07a8655f1db9a60374f9f8",
"https://bcr.bazel.build/modules/stardoc/0.5.3/MODULE.bazel": "c7f6948dae6999bf0db32c1858ae345f112cacf98f174c7a8bb707e41b974f1c", "https://bcr.bazel.build/modules/stardoc/0.5.3/MODULE.bazel": "c7f6948dae6999bf0db32c1858ae345f112cacf98f174c7a8bb707e41b974f1c",
"https://bcr.bazel.build/modules/stardoc/0.5.4/MODULE.bazel": "6569966df04610b8520957cb8e97cf2e9faac2c0309657c537ab51c16c18a2a4", "https://bcr.bazel.build/modules/stardoc/0.5.4/MODULE.bazel": "6569966df04610b8520957cb8e97cf2e9faac2c0309657c537ab51c16c18a2a4",
@ -2811,89 +2814,89 @@
"@@rules_zig+//zig:extensions.bzl%zig": { "@@rules_zig+//zig:extensions.bzl%zig": {
"general": { "general": {
"bzlTransitiveDigest": "wV3euhBZSI0D2ZlqbY1YW592mghFY88d9H6zBMQRaJY=", "bzlTransitiveDigest": "wV3euhBZSI0D2ZlqbY1YW592mghFY88d9H6zBMQRaJY=",
"usagesDigest": "FR3I3Uy7g7MHBta/AuDtKPMdikDVo0hgQZNdLunZDX4=", "usagesDigest": "PFS+UEa30JWWU52AUTNHD1nDzqpwjSts6jIKzD7gVJU=",
"recordedFileInputs": { "recordedFileInputs": {
"@@//bazel/zig_index.json": "42ea73420b15741770d5d8d0ded1311a2dfda1f963f0802dea3418f6e060ffef", "@@//bazel/zig_index.json": "e6c6b29fa227fa9b1806dc03af2b6dab00f52f34d39878557074e1626027259f",
"@@rules_zig+//zig/private/versions.json": "6eb85ebaee72c4e6fbc82d8ad73f69e0899d839b70a7e19caa792ab6a5fa0c36" "@@rules_zig+//zig/private/versions.json": "6eb85ebaee72c4e6fbc82d8ad73f69e0899d839b70a7e19caa792ab6a5fa0c36"
}, },
"recordedDirentsInputs": {}, "recordedDirentsInputs": {},
"envVariables": {}, "envVariables": {},
"generatedRepoSpecs": { "generatedRepoSpecs": {
"zig_0.15.1_aarch64-linux": { "zig_0.15.2_aarch64-linux": {
"repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository", "repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository",
"attributes": { "attributes": {
"url": "https://ziglang.org/download/0.15.1/zig-aarch64-linux-0.15.1.tar.xz", "url": "https://ziglang.org/download/0.15.2/zig-aarch64-linux-0.15.2.tar.xz",
"mirrors": [ "mirrors": [
"https://mirror.zml.ai/zig", "https://mirror.zml.ai/zig",
"https://ziglang.org/builds/" "https://ziglang.org/builds/"
], ],
"sha256": "bb4a8d2ad735e7fba764c497ddf4243cb129fece4148da3222a7046d3f1f19fe", "sha256": "958ed7d1e00d0ea76590d27666efbf7a932281b3d7ba0c6b01b0ff26498f667f",
"zig_version": "0.15.1", "zig_version": "0.15.2",
"platform": "aarch64-linux" "platform": "aarch64-linux"
} }
}, },
"zig_0.15.1_aarch64-macos": { "zig_0.15.2_aarch64-macos": {
"repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository", "repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository",
"attributes": { "attributes": {
"url": "https://ziglang.org/download/0.15.1/zig-aarch64-macos-0.15.1.tar.xz", "url": "https://ziglang.org/download/0.15.2/zig-aarch64-macos-0.15.2.tar.xz",
"mirrors": [ "mirrors": [
"https://mirror.zml.ai/zig", "https://mirror.zml.ai/zig",
"https://ziglang.org/builds/" "https://ziglang.org/builds/"
], ],
"sha256": "c4bd624d901c1268f2deb9d8eb2d86a2f8b97bafa3f118025344242da2c54d7b", "sha256": "3cc2bab367e185cdfb27501c4b30b1b0653c28d9f73df8dc91488e66ece5fa6b",
"zig_version": "0.15.1", "zig_version": "0.15.2",
"platform": "aarch64-macos" "platform": "aarch64-macos"
} }
}, },
"zig_0.15.1_aarch64-windows": { "zig_0.15.2_aarch64-windows": {
"repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository", "repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository",
"attributes": { "attributes": {
"url": "https://ziglang.org/download/0.15.1/zig-aarch64-windows-0.15.1.zip", "url": "https://ziglang.org/download/0.15.2/zig-aarch64-windows-0.15.2.zip",
"mirrors": [ "mirrors": [
"https://mirror.zml.ai/zig", "https://mirror.zml.ai/zig",
"https://ziglang.org/builds/" "https://ziglang.org/builds/"
], ],
"sha256": "1f1bf16228b0ffcc882b713dc5e11a6db4219cb30997e13c72e8e723c2104ec6", "sha256": "b926465f8872bf983422257cd9ec248bb2b270996fbe8d57872cca13b56fc370",
"zig_version": "0.15.1", "zig_version": "0.15.2",
"platform": "aarch64-windows" "platform": "aarch64-windows"
} }
}, },
"zig_0.15.1_x86_64-linux": { "zig_0.15.2_x86_64-linux": {
"repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository", "repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository",
"attributes": { "attributes": {
"url": "https://ziglang.org/download/0.15.1/zig-x86_64-linux-0.15.1.tar.xz", "url": "https://ziglang.org/download/0.15.2/zig-x86_64-linux-0.15.2.tar.xz",
"mirrors": [ "mirrors": [
"https://mirror.zml.ai/zig", "https://mirror.zml.ai/zig",
"https://ziglang.org/builds/" "https://ziglang.org/builds/"
], ],
"sha256": "c61c5da6edeea14ca51ecd5e4520c6f4189ef5250383db33d01848293bfafe05", "sha256": "02aa270f183da276e5b5920b1dac44a63f1a49e55050ebde3aecc9eb82f93239",
"zig_version": "0.15.1", "zig_version": "0.15.2",
"platform": "x86_64-linux" "platform": "x86_64-linux"
} }
}, },
"zig_0.15.1_x86_64-macos": { "zig_0.15.2_x86_64-macos": {
"repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository", "repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository",
"attributes": { "attributes": {
"url": "https://ziglang.org/download/0.15.1/zig-x86_64-macos-0.15.1.tar.xz", "url": "https://ziglang.org/download/0.15.2/zig-x86_64-macos-0.15.2.tar.xz",
"mirrors": [ "mirrors": [
"https://mirror.zml.ai/zig", "https://mirror.zml.ai/zig",
"https://ziglang.org/builds/" "https://ziglang.org/builds/"
], ],
"sha256": "9919392e0287cccc106dfbcbb46c7c1c3fa05d919567bb58d7eb16bca4116184", "sha256": "375b6909fc1495d16fc2c7db9538f707456bfc3373b14ee83fdd3e22b3d43f7f",
"zig_version": "0.15.1", "zig_version": "0.15.2",
"platform": "x86_64-macos" "platform": "x86_64-macos"
} }
}, },
"zig_0.15.1_x86_64-windows": { "zig_0.15.2_x86_64-windows": {
"repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository", "repoRuleId": "@@rules_zig+//zig/private/repo:zig_repository.bzl%zig_repository",
"attributes": { "attributes": {
"url": "https://ziglang.org/download/0.15.1/zig-x86_64-windows-0.15.1.zip", "url": "https://ziglang.org/download/0.15.2/zig-x86_64-windows-0.15.2.zip",
"mirrors": [ "mirrors": [
"https://mirror.zml.ai/zig", "https://mirror.zml.ai/zig",
"https://ziglang.org/builds/" "https://ziglang.org/builds/"
], ],
"sha256": "91e69e887ca8c943ce9a515df3af013d95a66a190a3df3f89221277ebad29e34", "sha256": "3a0ed1e8799a2f8ce2a6e6290a9ff22e6906f8227865911fb7ddedc3cc14cb0c",
"zig_version": "0.15.1", "zig_version": "0.15.2",
"platform": "x86_64-windows" "platform": "x86_64-windows"
} }
}, },
@ -2901,28 +2904,28 @@
"repoRuleId": "@@rules_zig+//zig/private/repo:toolchains_repo.bzl%toolchains_repo", "repoRuleId": "@@rules_zig+//zig/private/repo:toolchains_repo.bzl%toolchains_repo",
"attributes": { "attributes": {
"names": [ "names": [
"zig_0.15.1_aarch64-linux", "zig_0.15.2_aarch64-linux",
"zig_0.15.1_aarch64-macos", "zig_0.15.2_aarch64-macos",
"zig_0.15.1_aarch64-windows", "zig_0.15.2_aarch64-windows",
"zig_0.15.1_x86_64-linux", "zig_0.15.2_x86_64-linux",
"zig_0.15.1_x86_64-macos", "zig_0.15.2_x86_64-macos",
"zig_0.15.1_x86_64-windows" "zig_0.15.2_x86_64-windows"
], ],
"labels": [ "labels": [
"@zig_0.15.1_aarch64-linux//:zig_toolchain", "@zig_0.15.2_aarch64-linux//:zig_toolchain",
"@zig_0.15.1_aarch64-macos//:zig_toolchain", "@zig_0.15.2_aarch64-macos//:zig_toolchain",
"@zig_0.15.1_aarch64-windows//:zig_toolchain", "@zig_0.15.2_aarch64-windows//:zig_toolchain",
"@zig_0.15.1_x86_64-linux//:zig_toolchain", "@zig_0.15.2_x86_64-linux//:zig_toolchain",
"@zig_0.15.1_x86_64-macos//:zig_toolchain", "@zig_0.15.2_x86_64-macos//:zig_toolchain",
"@zig_0.15.1_x86_64-windows//:zig_toolchain" "@zig_0.15.2_x86_64-windows//:zig_toolchain"
], ],
"zig_versions": [ "zig_versions": [
"0.15.1", "0.15.2",
"0.15.1", "0.15.2",
"0.15.1", "0.15.2",
"0.15.1", "0.15.2",
"0.15.1", "0.15.2",
"0.15.1" "0.15.2"
], ],
"exec_lengths": [ "exec_lengths": [
2, 2,

2
runtimes/cuda/BUILD.bazel
View File

@ -5,7 +5,7 @@ zig_shared_library(
name = "zmlxcuda", name = "zmlxcuda",
# Use Clang's compiler-rt, but disable stack checking # Use Clang's compiler-rt, but disable stack checking
# to avoid requiring on the _zig_probe_stack symbol. # to avoid requiring on the _zig_probe_stack symbol.
zigopts = ["-fno-stack-check", "-fllvm"], zigopts = ["-fno-stack-check"],
main = "zmlxcuda.zig", main = "zmlxcuda.zig",
shared_lib_name = "libzmlxcuda.so.0", shared_lib_name = "libzmlxcuda.so.0",
visibility = ["@libpjrt_cuda//:__subpackages__"], visibility = ["@libpjrt_cuda//:__subpackages__"],

4
upb/BUILD.bazel
View File

@ -14,4 +14,8 @@ zig_library(
main = "upb.zig", main = "upb.zig",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [":upb_c"], deps = [":upb_c"],
zigopts = [
#TODO(cerisier): Remove me when this is done inside rules_zig.
"-fllvm",
],
) )

7
zml/BUILD.bazel
View File

@ -54,7 +54,12 @@ zig_library(
"torch.zig", "torch.zig",
"zml.zig", "zml.zig",
], ],
zigopts = ["-lc", "-freference-trace=20"], zigopts = [
"-lc",
"-freference-trace=20",
#TODO(cerisier): Remove me when this is done inside rules_zig.
"-fllvm",
],
main = "zml.zig", main = "zml.zig",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [ deps = [

4
zml/tokenizer/BUILD.bazel
View File

@ -19,6 +19,10 @@ zig_binary(
name = "main", name = "main",
main = "main.zig", main = "main.zig",
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
zigopts = [
#TODO(cerisier): Remove me when this is done inside rules_zig.
"-fllvm"
],
deps = [ deps = [
":tokenizer", ":tokenizer",
"//async", "//async",