Tarry Singh
9488672d4b
Also: - Bump XLA deps : `com_github_grpc_grpc` and `com_google_protobuf` - Inject `rules_ml_toolchain` - Fix `zig_proto_library` rule
18 lines
467 B
Python
18 lines
467 B
Python
load("//third_party/llvm:workspace.bzl", llvm = "repo")
|
|
load("//third_party/stablehlo:workspace.bzl", stablehlo = "repo")
|
|
load("//third_party/triton:workspace.bzl", triton = "repo")
|
|
|
|
def _xla_impl(mctx):
|
|
triton()
|
|
llvm("llvm-raw")
|
|
stablehlo()
|
|
return mctx.extension_metadata(
|
|
reproducible = True,
|
|
root_module_direct_deps = "all",
|
|
root_module_direct_dev_deps = [],
|
|
)
|
|
|
|
xla = module_extension(
|
|
implementation = _xla_impl,
|
|
)
|