WORKSPACE

workspace(name = "ml_models_public")

local_repository(
    name = "kelvin_sw",
    path = "../../sw/kelvin",
)

local_repository(
    name = "matcha",
    path = "../../hw/matcha",
)

local_repository(
    name = "lowrisc_opentitan",
    path = "../../hw/opentitan-upstream",
)

load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
http_archive(
    name = "rules_python",
    sha256 = "9d04041ac92a0985e344235f5d946f71ac543f1b1565f2cdbc9a2aaee8adf55b",
    strip_prefix = "rules_python-0.26.0",
    url = "https://github.com/bazelbuild/rules_python/releases/download/0.26.0/rules_python-0.26.0.tar.gz",
)

load("@rules_python//python:repositories.bzl", "py_repositories")

py_repositories()

load("@rules_python//python:repositories.bzl", "python_register_toolchains")

python_register_toolchains(
    name = "python3",
    python_version = "3.9",
)

load("@kelvin_sw//build_tools/bazel:repos.bzl", "kelvin_sw_repos")

kelvin_sw_repos()

# Register Kelvin toolchain
load("@kelvin_sw//platforms:registration.bzl", "kelvin_register_toolchain")

kelvin_register_toolchain()

load("@lowrisc_opentitan//third_party/crt:repos.bzl", lowrisc_crt_repos = "crt_repos")

lowrisc_crt_repos()

load("@crt//:repos.bzl", "crt_repos")

crt_repos()

load("@crt//config:registration.bzl", "crt_register_toolchains")

crt_register_toolchains(riscv32 = True)

load("@kelvin_sw//build_tools/bazel:repos.bzl", "tflm_repos")

tflm_repos()

load("@tflite-micro//tensorflow:workspace.bzl", "tf_repositories")

tf_repositories()

load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")

rules_pkg_dependencies()

load("@rules_python//python:pip.bzl", "pip_parse")

pip_parse(
    name = "ot_python_deps",
    requirements_lock = "//:opentitan_sanitized_requirements.txt",
    python_interpreter_target = "@python3_x86_64-unknown-linux-gnu//:python"
)

load("@ot_python_deps//:requirements.bzl", install_ot_python_deps = "install_deps")

install_ot_python_deps()

load("@lowrisc_opentitan//third_party/google:repos.bzl", "google_repos")

google_repos()

load("@lowrisc_opentitan//third_party/google:deps.bzl", "google_deps")

google_deps()

pip_parse(
    name = "tflm_pip_deps",
    requirements_lock = "@tflite-micro//third_party:python_requirements.txt",
    python_interpreter_target = "@python3_x86_64-unknown-linux-gnu//:python"
)

load("@tflm_pip_deps//:requirements.bzl", install_tflm_pip_deps = "install_deps")

install_tflm_pip_deps()

http_archive(
    name = "mlperf_tiny",
    strip_prefix = "tiny-debd5310ffbad653932e2995816e853197116dd9",
    urls = ["https://github.com/mlcommons/tiny/archive/debd5310ffbad653932e2995816e853197116dd9.zip"],
    sha256 = "e74de33f20aabfa587d0fa35a0cbe9a1bda6a9d82475b3947bcc588142f49414",
    build_file = "//third_party/mlperf_tiny:BUILD.mlperf_tiny",
)