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", |
Marat Dukhan | 0952f41 | 2020-04-13 02:04:41 -0700 | [diff] [blame] | 29 | strip_prefix = "FP16-3c54eacb74f6f5e39077300c5564156c424d77ba", |
| 30 | sha256 = "0d56bb92f649ec294dbccb13e04865e3c82933b6f6735d1d7145de45da700156", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 31 | urls = [ |
Marat Dukhan | 0952f41 | 2020-04-13 02:04:41 -0700 | [diff] [blame] | 32 | "https://github.com/Maratyszcza/FP16/archive/3c54eacb74f6f5e39077300c5564156c424d77ba.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 | 79eda12 | 2020-04-11 17:02:41 -0700 | [diff] [blame] | 40 | strip_prefix = "FXdiv-f7dd0576a1c8289ef099d4fd8b136b1c4487a873", |
| 41 | sha256 = "6e4b6e3c58e67c3bb090e286c4f235902c89b98cf3e67442a18f9167963aa286", |
| 42 | urls = ["https://github.com/Maratyszcza/FXdiv/archive/f7dd0576a1c8289ef099d4fd8b136b1c4487a873.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", |
Marat Dukhan | ce2d5b3 | 2020-04-14 19:05:15 -0700 | [diff] [blame] | 48 | strip_prefix = "pthreadpool-a61ed1ab70389c62f6f699ca1a30a2421d3ea594", |
| 49 | sha256 = "27c039e73846d0bdfe393833e91afafe45e61ba792cc60e1c62808090554ce4d", |
| 50 | urls = ["https://github.com/Maratyszcza/pthreadpool/archive/a61ed1ab70389c62f6f699ca1a30a2421d3ea594.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 | |
| 64 | # cpuinfo library, used for detecting processor characteristics |
| 65 | http_archive( |
| 66 | name = "cpuinfo", |
Marat Dukhan | bd18b51 | 2020-03-25 15:29:30 -0700 | [diff] [blame] | 67 | strip_prefix = "cpuinfo-0cc563acb9baac39f2c1349bc42098c4a1da59e3", |
| 68 | sha256 = "80625d0b69a3d69b70c2236f30db2c542d0922ccf9bb51a61bc39c49fac91a35", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 69 | urls = [ |
Marat Dukhan | bd18b51 | 2020-03-25 15:29:30 -0700 | [diff] [blame] | 70 | "https://github.com/pytorch/cpuinfo/archive/0cc563acb9baac39f2c1349bc42098c4a1da59e3.tar.gz", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 71 | ], |
| 72 | build_file = "@//third_party:cpuinfo.BUILD", |
| 73 | ) |
| 74 | |
| 75 | # psimd library, used for fallback 128-bit SIMD micro-kernels |
| 76 | http_archive( |
| 77 | name = "psimd", |
Marat Dukhan | 3ba868c | 2020-02-21 09:08:37 -0800 | [diff] [blame] | 78 | strip_prefix = "psimd-88882f601f8179e1987b7e7cf4a8012c9080ad44", |
| 79 | sha256 = "c621f9bb1ff9ab8f0fa4a04f3239d13b345a6e865318d7b464aa80531a1abb2c", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 80 | urls = [ |
Marat Dukhan | 3ba868c | 2020-02-21 09:08:37 -0800 | [diff] [blame] | 81 | "https://github.com/Maratyszcza/psimd/archive/88882f601f8179e1987b7e7cf4a8012c9080ad44.tar.gz", |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 82 | ], |
| 83 | build_file = "@//third_party:psimd.BUILD", |
| 84 | ) |
| 85 | |
Marat Dukhan | bb0c2f8 | 2020-03-25 13:31:19 -0700 | [diff] [blame] | 86 | # Ruy library, used to benchmark against |
| 87 | http_archive( |
Benoit Jacob | b038fdc | 2020-03-25 12:14:20 -0700 | [diff] [blame] | 88 | name = "ruy", |
Benoit Jacob | b026e22 | 2020-04-16 12:30:03 -0700 | [diff] [blame^] | 89 | strip_prefix = "ruy-9f53ba413e6fc879236dcaa3e008915973d67a4f", |
| 90 | sha256 = "fe8345f521bb378745ebdd0f8c5937414849936851d2ec2609774eb2d7098e54", |
Benoit Jacob | b038fdc | 2020-03-25 12:14:20 -0700 | [diff] [blame] | 91 | urls = [ |
Benoit Jacob | b026e22 | 2020-04-16 12:30:03 -0700 | [diff] [blame^] | 92 | "https://github.com/google/ruy/archive/9f53ba413e6fc879236dcaa3e008915973d67a4f.zip", |
Benoit Jacob | b038fdc | 2020-03-25 12:14:20 -0700 | [diff] [blame] | 93 | ], |
| 94 | ) |
| 95 | |
Marat Dukhan | 08c4a43 | 2019-10-03 09:29:21 -0700 | [diff] [blame] | 96 | # Android NDK location and version is auto-detected from $ANDROID_NDK_HOME environment variable |
| 97 | android_ndk_repository(name = "androidndk") |
| 98 | |
| 99 | # Android SDK location and API is auto-detected from $ANDROID_HOME environment variable |
| 100 | android_sdk_repository(name = "androidsdk") |