Radix/pjrt/BUILD.bazel

load("@rules_zig//zig:defs.bzl", "zig_library")

zig_library(
    name = "pjrt",
    srcs = ["ffi.zig"],
    main = "pjrt.zig",
    visibility = ["//visibility:public"],
    deps = [
        "//stdx",
        "@xla//xla/ffi/api:c_api",
        "@xla//xla/pjrt/c:pjrt_c_api_ffi_extension_hdrs",
        "@xla//xla/pjrt/c:pjrt_c_api_gpu_extension_hdrs",
        "@xla//xla/pjrt/c:pjrt_c_api_hdrs",
        "@xla//xla/pjrt/c:pjrt_c_api_triton_extension_hdrs",
    ],
)
Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML. 2023-01-02 14:28:25 +00:00			`load("@rules_zig//zig:defs.bzl", "zig_library")`

			`zig_library(`
			`name = "pjrt",`
pjrt,zml: remove profiler Closes Progress towards 2025-06-06 12:24:42 +00:00			`srcs = ["ffi.zig"],`
Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML. 2023-01-02 14:28:25 +00:00			`main = "pjrt.zig",`
			`visibility = ["//visibility:public"],`
			`deps = [`
Update workspace dependencies to newer LLVM, XLA, StableHLO, and PJRT versions and expose new pjrt plugin attribute APIs and stablehlo version APIs in build and runtime configurations. 2023-08-07 12:28:36 +00:00			`"//stdx",`
pjrt: add FFI bindings for custom calls 2024-09-10 09:14:28 +00:00			`"@xla//xla/ffi/api:c_api",`
			`"@xla//xla/pjrt/c:pjrt_c_api_ffi_extension_hdrs",`
Add initial Bazel build configuration, async runtime implementation, and core MLIR dialect definitions for ZML. 2023-01-02 14:28:25 +00:00			`"@xla//xla/pjrt/c:pjrt_c_api_gpu_extension_hdrs",`
			`"@xla//xla/pjrt/c:pjrt_c_api_hdrs",`
pjrt: add FFI bindings for custom calls 2024-09-10 09:14:28 +00:00			`"@xla//xla/pjrt/c:pjrt_c_api_triton_extension_hdrs",`
Update rules_zig: add zig_srcs target, fix source handling bug, clean up BUILD files, adjust async/coro.zig tests, and disable nemo and yaml model loaders. 2025-03-13 12:27:21 +00:00			`],`
			`)`