Tarry Singh
1427286716
This PR fixes the neuron runtime with the following: Proxy the PJRT Api method to enforce the client struct sizes since the neuron PJRT plugin doesn't use `>=` but `==` to assert them, breaking PJRT compatibility guarantees. Fixes https://github.com/aws-neuron/aws-neuron-sdk/issues/1095 Reimplement `libneuronxla` in Zig to control neuronx-cc sandboxing and invocation. Implement a python bootstrapper in Zig to create a full blown `neuronx-cc` executable, avoiding the infamous chicken and egg problem of python executables boostrapping when sandboxed (due to fixed path shebangs). --------- Co-authored-by: Corentin Kerisit <corentin.kerisit@gmail.com>
17 lines
296 B
Python
17 lines
296 B
Python
load("@rules_zig//zig:defs.bzl", "zig_library")
|
|
|
|
cc_library(
|
|
name = "upb_c",
|
|
hdrs = ["upb.h"],
|
|
deps = [
|
|
"@com_google_protobuf//upb/message:copy",
|
|
],
|
|
)
|
|
|
|
zig_library(
|
|
name = "upb",
|
|
main = "upb.zig",
|
|
deps = [":upb_c"],
|
|
visibility = ["//visibility:public"],
|
|
)
|