blob: c1908ba227d1773ae2e8134b14f0a1b2ed5d279e [file] [log] [blame]
workspace(name = "xnnpack")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
# Bazel rule definitions
http_archive(
name = "rules_cc",
strip_prefix = "rules_cc-master",
urls = ["https://github.com/bazelbuild/rules_cc/archive/master.zip"],
)
# Google Test framework, used by most unit-tests.
http_archive(
name = "com_google_googletest",
strip_prefix = "googletest-master",
urls = ["https://github.com/google/googletest/archive/master.zip"],
)
# Google Benchmark library, used in micro-benchmarks.
http_archive(
name = "com_google_benchmark",
strip_prefix = "benchmark-master",
urls = ["https://github.com/google/benchmark/archive/master.zip"],
)
# FP16 library, used for half-precision conversions
http_archive(
name = "FP16",
strip_prefix = "FP16-3c54eacb74f6f5e39077300c5564156c424d77ba",
sha256 = "0d56bb92f649ec294dbccb13e04865e3c82933b6f6735d1d7145de45da700156",
urls = [
"https://github.com/Maratyszcza/FP16/archive/3c54eacb74f6f5e39077300c5564156c424d77ba.zip",
],
build_file = "@//third_party:FP16.BUILD",
)
# FXdiv library, used for repeated integer division by the same factor
http_archive(
name = "FXdiv",
strip_prefix = "FXdiv-b408327ac2a15ec3e43352421954f5b1967701d1",
sha256 = "ab7dfb08829bee33dca38405d647868fb214ac685e379ec7ef2bebcd234cd44d",
urls = ["https://github.com/Maratyszcza/FXdiv/archive/b408327ac2a15ec3e43352421954f5b1967701d1.zip"],
)
# pthreadpool library, used for parallelization
http_archive(
name = "pthreadpool",
strip_prefix = "pthreadpool-e1642461b3b0217d23d6664d839a060f54e4e652",
sha256 = "c4d4a16053ec0e5125dbd8ae1d6d2ba99601d6fdcf8601a0d51d02a048c40348",
urls = ["https://github.com/Maratyszcza/pthreadpool/archive/e1642461b3b0217d23d6664d839a060f54e4e652.zip"],
)
# clog library, used for logging
http_archive(
name = "clog",
strip_prefix = "cpuinfo-d5e37adf1406cf899d7d9ec1d317c47506ccb970",
sha256 = "3f2dc1970f397a0e59db72f9fca6ff144b216895c1d606f6c94a507c1e53a025",
urls = [
"https://github.com/pytorch/cpuinfo/archive/d5e37adf1406cf899d7d9ec1d317c47506ccb970.tar.gz",
],
build_file = "@//third_party:clog.BUILD",
)
# cpuinfo library, used for detecting processor characteristics
http_archive(
name = "cpuinfo",
strip_prefix = "cpuinfo-6cecd15784fcb6c5c0aa7311c6248879ce2cb8b2",
sha256 = "b1f2ee97e46d8917a66bcb47452fc510d511829556c93b83e06841b9b35261a5",
urls = [
"https://github.com/pytorch/cpuinfo/archive/6cecd15784fcb6c5c0aa7311c6248879ce2cb8b2.zip",
],
build_file = "@//third_party:cpuinfo.BUILD",
patches = ["@//third_party:cpuinfo.patch"],
)
# psimd library, used for fallback 128-bit SIMD micro-kernels
http_archive(
name = "psimd",
strip_prefix = "psimd-072586a71b55b7f8c584153d223e95687148a900",
sha256 = "dc615342bcbe51ca885323e51b68b90ed9bb9fa7df0f4419dbfa0297d5e837b7",
urls = ["https://github.com/Maratyszcza/psimd/archive/072586a71b55b7f8c584153d223e95687148a900.zip"],
build_file = "@//third_party:psimd.BUILD",
)
# Ruy library, used to benchmark against
http_archive(
name = "ruy",
strip_prefix = "ruy-9f53ba413e6fc879236dcaa3e008915973d67a4f",
sha256 = "fe8345f521bb378745ebdd0f8c5937414849936851d2ec2609774eb2d7098e54",
urls = [
"https://github.com/google/ruy/archive/9f53ba413e6fc879236dcaa3e008915973d67a4f.zip",
],
)
# Android NDK location and version is auto-detected from $ANDROID_NDK_HOME environment variable
android_ndk_repository(name = "androidndk")
# Android SDK location and API is auto-detected from $ANDROID_HOME environment variable
android_sdk_repository(name = "androidsdk")