Marat Dukhan | 7ae046d | 2017-02-16 16:57:21 -0500 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | |
| 3 | |
| 4 | import confu |
| 5 | parser = confu.standard_parser("FP16 configuration script") |
Marat Dukhan | 1718dae | 2017-02-17 22:41:54 -0500 | [diff] [blame] | 6 | parser.add_argument("--compare", dest="compare", action="store_true", |
| 7 | help="Enable performance comparison with other half-precision implementations") |
Marat Dukhan | 7ae046d | 2017-02-16 16:57:21 -0500 | [diff] [blame] | 8 | |
| 9 | def main(args): |
| 10 | options = parser.parse_args(args) |
| 11 | build = confu.Build.from_options(options) |
| 12 | |
| 13 | build.export_cpath("include", ["fp16.h"]) |
| 14 | |
Marat Dukhan | 7f1c3cf | 2017-02-16 22:42:06 -0500 | [diff] [blame] | 15 | with build.options(source_dir="test", extra_include_dirs="test", deps=[build.deps.googletest, build.deps.psimd]): |
Marat Dukhan | 7ae046d | 2017-02-16 16:57:21 -0500 | [diff] [blame] | 16 | fp16_tables = build.cxx("tables.cc") |
Marat Dukhan | 7ae046d | 2017-02-16 16:57:21 -0500 | [diff] [blame] | 17 | build.unittest("ieee-to-fp32-bits", |
| 18 | [build.cxx("ieee-to-fp32-bits.cc"), fp16_tables]) |
| 19 | build.unittest("ieee-to-fp32-value", |
| 20 | [build.cxx("ieee-to-fp32-value.cc"), fp16_tables]) |
Marat Dukhan | 7f1c3cf | 2017-02-16 22:42:06 -0500 | [diff] [blame] | 21 | build.unittest("ieee-from-fp32-value", |
| 22 | [build.cxx("ieee-from-fp32-value.cc"), fp16_tables]) |
Marat Dukhan | 7ae046d | 2017-02-16 16:57:21 -0500 | [diff] [blame] | 23 | |
Marat Dukhan | 7ae046d | 2017-02-16 16:57:21 -0500 | [diff] [blame] | 24 | build.unittest("alt-to-fp32-bits", |
| 25 | [build.cxx("alt-to-fp32-bits.cc"), fp16_tables]) |
| 26 | build.unittest("alt-to-fp32-value", |
| 27 | [build.cxx("alt-to-fp32-value.cc"), fp16_tables]) |
Marat Dukhan | 7f5fc4a | 2017-02-16 17:17:49 -0500 | [diff] [blame] | 28 | build.unittest("alt-from-fp32-value", |
| 29 | [build.cxx("alt-from-fp32-value.cc"), fp16_tables]) |
| 30 | |
Marat Dukhan | bec1bf1 | 2017-02-22 04:23:14 -0500 | [diff] [blame] | 31 | if build.target.is_x86_64: |
| 32 | stubs = build.peachpy("peachpy/stubs.py") |
| 33 | build.unittest("alt-xmm-to-fp32-ymm-avx", [build.cxx("peachpy/alt-xmm-to-fp32-xmm-avx.cc"), stubs]) |
| 34 | build.unittest("alt-xmm-to-fp32-ymm-avx2", [build.cxx("peachpy/alt-xmm-to-fp32-ymm-avx2.cc"), stubs]) |
| 35 | |
Marat Dukhan | 09c7b38 | 2017-02-17 15:15:46 -0500 | [diff] [blame] | 36 | if not build.target.is_emscripten: |
| 37 | build.unittest("ieee-to-fp32-psimd", build.cxx("ieee-to-fp32-psimd.cc")) |
| 38 | build.unittest("alt-to-fp32-psimd", build.cxx("alt-to-fp32-psimd.cc")) |
Marat Dukhan | 7f1c3cf | 2017-02-16 22:42:06 -0500 | [diff] [blame] | 39 | |
Marat Dukhan | 09c7b38 | 2017-02-17 15:15:46 -0500 | [diff] [blame] | 40 | build.unittest("ieee-to-fp32x2-psimd", build.cxx("ieee-to-fp32x2-psimd.cc")) |
| 41 | build.unittest("alt-to-fp32x2-psimd", build.cxx("alt-to-fp32x2-psimd.cc")) |
Marat Dukhan | a0e32e8 | 2017-02-17 02:57:47 -0500 | [diff] [blame] | 42 | |
Marat Dukhan | 7f5fc4a | 2017-02-16 17:17:49 -0500 | [diff] [blame] | 43 | build.unittest("bitcasts", build.cxx("bitcasts.cc")) |
| 44 | |
Marat Dukhan | a8f140a | 2017-03-23 11:44:07 +0000 | [diff] [blame] | 45 | macros = ["BENCHMARK_HAS_NO_INLINE_ASSEMBLY"] |
| 46 | if options.compare: |
| 47 | macros.append("FP16_COMPARATIVE_BENCHMARKS") |
| 48 | with build.options(source_dir="bench", extra_include_dirs=".", macros=macros, |
Marat Dukhan | 1718dae | 2017-02-17 22:41:54 -0500 | [diff] [blame] | 49 | deps=[build.deps.googlebenchmark, build.deps.psimd]): |
| 50 | |
Marat Dukhan | e54a2ce | 2017-02-17 01:24:56 -0500 | [diff] [blame] | 51 | build.benchmark("ieee-element-bench", build.cxx("ieee-element.cc")) |
| 52 | build.benchmark("alt-element-bench", build.cxx("alt-element.cc")) |
| 53 | |
Marat Dukhan | 1718dae | 2017-02-17 22:41:54 -0500 | [diff] [blame] | 54 | build.benchmark("from-ieee-array-bench", build.cxx("from-ieee-array.cc")) |
| 55 | build.benchmark("from-alt-array-bench", build.cxx("from-alt-array.cc")) |
| 56 | |
| 57 | build.benchmark("to-ieee-array-bench", build.cxx("to-ieee-array.cc")) |
| 58 | build.benchmark("to-alt-array-bench", build.cxx("to-alt-array.cc")) |
Marat Dukhan | 7ae046d | 2017-02-16 16:57:21 -0500 | [diff] [blame] | 59 | |
| 60 | return build |
| 61 | |
| 62 | |
| 63 | if __name__ == "__main__": |
| 64 | import sys |
| 65 | main(sys.argv[1:]).generate() |