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

cc_library(
    name = "tpuinfo",
    srcs = ["tpuinfo.cc"],
    hdrs = ["tpuinfo.h"],
    deps = [
        "@cloud_accelerator_diagnostics//:tpu_metric_service_cc_grpc",
        "@grpc//:grpc++_unsecure",
    ],
)

zig_library(
    name = "tpu",
    import_name = "zml-smi/platforms/tpu",
    main = "metrics.zig",
    srcs = [
        "process.zig",
        "tpuinfo.zig",
    ],
    visibility = ["//visibility:public"],
    deps = [
        ":tpuinfo",
        "//bin/zml-smi/collector",
        "//bin/zml-smi/info",
        "//bin/zml-smi/utils:double_buffer",
        "//bin/zml-smi/utils:sysfs",
    ],
)