# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")

cc_library(
    name = "tosa_checker_lib",
    srcs = ["tosa_checker.cc"],
    hdrs = ["tosa_checker.h"],
    deps = [
        "@llvm-project//mlir:MlirTranslateMain",
        "@org_tensorflow//tensorflow/compiler/mlir/lite:flatbuffer_translate_lib",
        "@org_tensorflow//tensorflow/compiler/mlir/tosa:tfl_passes",
        "@pybind11",
    ],
)

pybind_extension(
    name = "_tosa_checker_wrapper",
    srcs = [
        "tosa_checker_pybind11.cc",
    ],
    deps = [
        ":tosa_checker_lib",
    ],
)

py_library(
    name = "tosa_checker",
    srcs = [
        "__init__.py",
    ],
    srcs_version = "PY3",
    visibility = ["//visibility:public"],
    data = ["//tosa_checker:_tosa_checker_wrapper.so"],
)