1 files changed, 35 insertions, 0 deletions
diff --git a/tosa_checker/BUILD b/tosa_checker/BUILD
new file mode 100644
index 0000000..8c1c32d
--- /dev/null
+++ b/tosa_checker/BUILD
@@ -0,0 +1,35 @@
+# 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"],
+)