2023-01-02 14:28:25 +00:00
|
|
|
load("@rules_zig//zig:defs.bzl", "zig_library")
|
|
|
|
|
load("//bazel:zig.bzl", "zig_cc_test")
|
|
|
|
|
|
|
|
|
|
zig_library(
|
|
|
|
|
name = "dialects",
|
|
|
|
|
srcs = [
|
|
|
|
|
"arith.zig",
|
|
|
|
|
"func.zig",
|
|
|
|
|
"math.zig",
|
2024-08-26 14:19:00 +00:00
|
|
|
"scf.zig",
|
2023-01-02 14:28:25 +00:00
|
|
|
"tensor.zig",
|
|
|
|
|
],
|
|
|
|
|
import_name = "mlir/dialects",
|
|
|
|
|
main = "dialects.zig",
|
|
|
|
|
visibility = ["//visibility:public"],
|
|
|
|
|
deps = [
|
|
|
|
|
":stablehlo",
|
|
|
|
|
"//mlir",
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
zig_cc_test(
|
|
|
|
|
name = "test",
|
|
|
|
|
deps = [":dialects"],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
zig_library(
|
|
|
|
|
name = "stablehlo",
|
|
|
|
|
import_name = "mlir/dialects/stablehlo",
|
|
|
|
|
main = "stablehlo.zig",
|
|
|
|
|
visibility = ["//mlir/dialects:__subpackages__"],
|
|
|
|
|
deps = [
|
|
|
|
|
"//mlir",
|
|
|
|
|
"//mlir:c",
|
2024-07-16 13:23:07 +00:00
|
|
|
"//stdx",
|
2024-05-31 13:02:46 +00:00
|
|
|
"@stablehlo//:stablehlo_dialect_capi",
|
2023-01-02 14:28:25 +00:00
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
zig_cc_test(
|
|
|
|
|
name = "stablehlo_test",
|
|
|
|
|
deps = [":stablehlo"],
|
|
|
|
|
)
|