34 lines
716 B
Python
34 lines
716 B
Python
load("@rules_zig//zig:defs.bzl", "zig_library")
|
|
|
|
cc_library(
|
|
name = "empty",
|
|
)
|
|
|
|
cc_library(
|
|
name = "libpjrt_cpu",
|
|
defines = ["ZML_RUNTIME_CPU"],
|
|
deps = select({
|
|
"@platforms//os:macos": ["@libpjrt_cpu_darwin_arm64//:libpjrt_cpu"],
|
|
"@platforms//os:linux": ["@libpjrt_cpu_linux_amd64//:libpjrt_cpu"],
|
|
}),
|
|
)
|
|
|
|
zig_library(
|
|
name = "cpu",
|
|
import_name = "runtimes/cpu",
|
|
main = "cpu.zig",
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"//pjrt",
|
|
] + select({
|
|
"//runtimes:cpu.enabled": [":libpjrt_cpu"],
|
|
"//conditions:default": [":empty"],
|
|
}),
|
|
)
|
|
|
|
filegroup(
|
|
name = "layers",
|
|
srcs = [],
|
|
visibility = ["//visibility:public"],
|
|
)
|