Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 1 | workspace(name = "xnnpack") |
| 2 | |
| 3 | load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive") |
| 4 | |
Marat Dukhan | 79eda12 | 2020-04-11 17:02:41 -0700 | [diff] [blame] | 5 | # Bazel rule definitions |
| 6 | http_archive( |
| 7 | name = "rules_cc", |
| 8 | strip_prefix = "rules_cc-master", |
| 9 | urls = ["https://github.com/bazelbuild/rules_cc/archive/master.zip"], |
| 10 | ) |
| 11 | |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 12 | # Google Test framework, used by most unit-tests. |
| 13 | http_archive( |
Marat Dukhan | 79eda12 | 2020-04-11 17:02:41 -0700 | [diff] [blame] | 14 | name = "com_google_googletest", |
| 15 | strip_prefix = "googletest-master", |
| 16 | urls = ["https://github.com/google/googletest/archive/master.zip"], |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 17 | ) |
| 18 | |
| 19 | # Google Benchmark library, used in micro-benchmarks. |
| 20 | http_archive( |
| 21 | name = "com_google_benchmark", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 22 | strip_prefix = "benchmark-master", |
Marat Dukhan | 79eda12 | 2020-04-11 17:02:41 -0700 | [diff] [blame] | 23 | urls = ["https://github.com/google/benchmark/archive/master.zip"], |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 24 | ) |
| 25 | |
| 26 | # FP16 library, used for half-precision conversions |
| 27 | http_archive( |
| 28 | name = "FP16", |
Jerry Shih | accc49a | 2021-03-25 00:57:26 -0700 | [diff] [blame] | 29 | strip_prefix = "FP16-0a92994d729ff76a58f692d3028ca1b64b145d91", |
| 30 | sha256 = "e66e65515fa09927b348d3d584c68be4215cfe664100d01c9dbc7655a5716d70", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 31 | urls = [ |
Jerry Shih | accc49a | 2021-03-25 00:57:26 -0700 | [diff] [blame] | 32 | "https://github.com/Maratyszcza/FP16/archive/0a92994d729ff76a58f692d3028ca1b64b145d91.zip", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 33 | ], |
| 34 | build_file = "@//third_party:FP16.BUILD", |
| 35 | ) |
| 36 | |
| 37 | # FXdiv library, used for repeated integer division by the same factor |
| 38 | http_archive( |
| 39 | name = "FXdiv", |
Marat Dukhan | 10a3808 | 2020-04-17 03:58:35 -0700 | [diff] [blame] | 40 | strip_prefix = "FXdiv-b408327ac2a15ec3e43352421954f5b1967701d1", |
| 41 | sha256 = "ab7dfb08829bee33dca38405d647868fb214ac685e379ec7ef2bebcd234cd44d", |
| 42 | urls = ["https://github.com/Maratyszcza/FXdiv/archive/b408327ac2a15ec3e43352421954f5b1967701d1.zip"], |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 43 | ) |
| 44 | |
| 45 | # pthreadpool library, used for parallelization |
| 46 | http_archive( |
| 47 | name = "pthreadpool", |
Simon Maurer | 4d2d3e5 | 2021-03-03 23:41:12 +0100 | [diff] [blame] | 48 | strip_prefix = "pthreadpool-b8374f80e42010941bda6c85b0e3f1a1bd77a1e0", |
| 49 | sha256 = "b96413b10dd8edaa4f6c0a60c6cf5ef55eebeef78164d5d69294c8173457f0ec", |
| 50 | urls = ["https://github.com/Maratyszcza/pthreadpool/archive/b8374f80e42010941bda6c85b0e3f1a1bd77a1e0.zip"], |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 51 | ) |
| 52 | |
| 53 | # clog library, used for logging |
| 54 | http_archive( |
| 55 | name = "clog", |
| 56 | strip_prefix = "cpuinfo-d5e37adf1406cf899d7d9ec1d317c47506ccb970", |
| 57 | sha256 = "3f2dc1970f397a0e59db72f9fca6ff144b216895c1d606f6c94a507c1e53a025", |
| 58 | urls = [ |
| 59 | "https://github.com/pytorch/cpuinfo/archive/d5e37adf1406cf899d7d9ec1d317c47506ccb970.tar.gz", |
| 60 | ], |
| 61 | build_file = "@//third_party:clog.BUILD", |
| 62 | ) |
| 63 | |
Marat Dukhan | 569583e | 2020-05-17 19:00:21 -0700 | [diff] [blame] | 64 | |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 65 | # cpuinfo library, used for detecting processor characteristics |
| 66 | http_archive( |
| 67 | name = "cpuinfo", |
Marat Dukhan | 0fff1bd | 2020-12-14 00:54:43 -0800 | [diff] [blame] | 68 | strip_prefix = "cpuinfo-ed8b86a253800bafdb7b25c5c399f91bff9cb1f3", |
| 69 | sha256 = "a7f9a188148a1660149878f737f42783e72f33a4f842f3e362fee2c981613e53", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 70 | urls = [ |
Marat Dukhan | 0fff1bd | 2020-12-14 00:54:43 -0800 | [diff] [blame] | 71 | "https://github.com/pytorch/cpuinfo/archive/ed8b86a253800bafdb7b25c5c399f91bff9cb1f3.zip", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 72 | ], |
| 73 | build_file = "@//third_party:cpuinfo.BUILD", |
Marat Dukhan | 54f5917 | 2020-04-25 14:59:26 -0700 | [diff] [blame] | 74 | patches = ["@//third_party:cpuinfo.patch"], |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 75 | ) |
| 76 | |
Marat Dukhan | bb0c2f8 | 2020-03-25 13:31:19 -0700 | [diff] [blame] | 77 | # Ruy library, used to benchmark against |
| 78 | http_archive( |
Benoit Jacob | b038fdc | 2020-03-25 12:14:20 -0700 | [diff] [blame] | 79 | name = "ruy", |
Benoit Jacob | b026e22 | 2020-04-16 12:30:03 -0700 | [diff] [blame] | 80 | strip_prefix = "ruy-9f53ba413e6fc879236dcaa3e008915973d67a4f", |
| 81 | sha256 = "fe8345f521bb378745ebdd0f8c5937414849936851d2ec2609774eb2d7098e54", |
Benoit Jacob | b038fdc | 2020-03-25 12:14:20 -0700 | [diff] [blame] | 82 | urls = [ |
Benoit Jacob | b026e22 | 2020-04-16 12:30:03 -0700 | [diff] [blame] | 83 | "https://github.com/google/ruy/archive/9f53ba413e6fc879236dcaa3e008915973d67a4f.zip", |
Benoit Jacob | b038fdc | 2020-03-25 12:14:20 -0700 | [diff] [blame] | 84 | ], |
| 85 | ) |
| 86 | |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 87 | # Android NDK location and version is auto-detected from $ANDROID_NDK_HOME environment variable |
| 88 | android_ndk_repository(name = "androidndk") |
| 89 | |
| 90 | # Android SDK location and API is auto-detected from $ANDROID_HOME environment variable |
| 91 | android_sdk_repository(name = "androidsdk") |