| # SPDX-FileCopyrightText: Copyright 2022-2023, 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", |
| "@org_tensorflow//tensorflow/compiler/mlir/tosa:tf_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"], |
| ) |