Michele Di Giorgio | d9eaf61 | 2020-07-08 11:12:57 +0100 | [diff] [blame] | 1 | # Copyright (c) 2017 Arm Limited. |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 2 | # |
| 3 | # SPDX-License-Identifier: MIT |
| 4 | # |
| 5 | # Permission is hereby granted, free of charge, to any person obtaining a copy |
| 6 | # of this software and associated documentation files (the "Software"), to |
| 7 | # deal in the Software without restriction, including without limitation the |
| 8 | # rights to use, copy, modify, merge, publish, distribute, sublicense, and/or |
| 9 | # sell copies of the Software, and to permit persons to whom the Software is |
| 10 | # furnished to do so, subject to the following conditions: |
| 11 | # |
| 12 | # The above copyright notice and this permission notice shall be included in all |
| 13 | # copies or substantial portions of the Software. |
| 14 | # |
| 15 | # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| 16 | # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| 17 | # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
| 18 | # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| 19 | # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
| 20 | # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
| 21 | # SOFTWARE. |
| 22 | import SCons |
| 23 | import os.path |
| 24 | |
| 25 | Import('env') |
Anthony Barbier | 01bbd5f | 2018-11-01 15:10:51 +0000 | [diff] [blame] | 26 | Import('install_bin') |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 27 | |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 28 | examples_env = env.Clone() |
| 29 | |
| 30 | examples_env.Append(CPPPATH = ["#"]) |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 31 | |
| 32 | # Build examples |
| 33 | utils = examples_env.Object("../utils/Utils.cpp") |
| 34 | |
Georgios Pinitas | 4551403 | 2020-12-30 00:03:09 +0000 | [diff] [blame] | 35 | if env['os'] in ['android', 'macos', 'bare_metal'] or env['standalone']: |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 36 | Import('arm_compute_graph_a') |
Pablo Tello | c6cb35a | 2017-06-21 15:39:47 +0100 | [diff] [blame] | 37 | Import('arm_compute_a') |
Anthony Barbier | b2881fc | 2017-09-29 17:12:12 +0100 | [diff] [blame] | 38 | Import('arm_compute_core_a') |
| 39 | arm_compute_libs = [ arm_compute_a, arm_compute_core_a ] |
Anthony Barbier | 8ddab14 | 2018-07-13 11:52:37 +0100 | [diff] [blame] | 40 | arm_compute_graph_libs = arm_compute_libs # The graph library needs to be linked separately with --whole-archive |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 41 | arm_compute_dependency = arm_compute_a |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 42 | graph_dependency = [arm_compute_graph_a] |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 43 | else: |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 44 | Import('arm_compute_graph_so') |
Pablo Tello | c6cb35a | 2017-06-21 15:39:47 +0100 | [diff] [blame] | 45 | Import('arm_compute_so') |
Anthony Barbier | 8ddab14 | 2018-07-13 11:52:37 +0100 | [diff] [blame] | 46 | arm_compute_libs = ["arm_compute", "arm_compute_core"] |
| 47 | arm_compute_graph_libs = [ "arm_compute_graph" ] + arm_compute_libs |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 48 | arm_compute_dependency = arm_compute_so |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 49 | graph_dependency = [arm_compute_graph_so] |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 50 | |
Michalis Spyrou | 6a9e801 | 2020-02-07 09:45:55 +0000 | [diff] [blame] | 51 | extra_link_flags = [] |
| 52 | if env['os'] != 'bare_metal': |
| 53 | extra_link_flags += ['-fstack-protector-strong'] |
| 54 | |
Georgios Pinitas | 4551403 | 2020-12-30 00:03:09 +0000 | [diff] [blame] | 55 | load_whole_archive = '-Wl,--whole-archive' |
| 56 | noload_whole_archive = '-Wl,--no-whole-archive' |
| 57 | if 'macos' in examples_env['os']: |
| 58 | load_whole_archive = '-Wl,-force_load' |
| 59 | noload_whole_archive = '-Wl,-noall_load' |
| 60 | |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 61 | # Build graph examples |
| 62 | graph_utils = examples_env.Object("../utils/GraphUtils.cpp") |
Georgios Pinitas | 12be7ab | 2018-07-03 12:06:23 +0100 | [diff] [blame] | 63 | graph_utils += examples_env.Object("../utils/CommonGraphOptions.cpp") |
Anthony Barbier | 7fb7b61 | 2018-05-16 11:58:52 +0100 | [diff] [blame] | 64 | examples_libs = examples_env.get("LIBS",[]) |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 65 | for file in Glob("./graph_*.cpp"): |
| 66 | example = os.path.basename(os.path.splitext(str(file))[0]) |
| 67 | prog = None |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 68 | |
Georgios Pinitas | 4551403 | 2020-12-30 00:03:09 +0000 | [diff] [blame] | 69 | if env['os'] in ['android', 'macos', 'bare_metal'] or env['standalone']: |
| 70 | prog = examples_env.Program(example, ["{}.cpp".format(example), utils, graph_utils], LIBS = examples_libs + arm_compute_graph_libs, LINKFLAGS=examples_env["LINKFLAGS"]+[load_whole_archive, graph_dependency, noload_whole_archive] + extra_link_flags) |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 71 | Depends(prog, graph_dependency) |
Michalis Spyrou | 165308c | 2019-04-10 10:10:21 +0100 | [diff] [blame] | 72 | prog = install_bin(prog) |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 73 | else: |
| 74 | #-Wl,--allow-shlib-undefined: Ignore dependencies of dependencies |
Anthony Barbier | 7fb7b61 | 2018-05-16 11:58:52 +0100 | [diff] [blame] | 75 | prog = examples_env.Program(example, ["{}.cpp".format(example), utils, graph_utils], LIBS = examples_libs + arm_compute_graph_libs, LINKFLAGS=examples_env["LINKFLAGS"]+['-Wl,--allow-shlib-undefined'] ) |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 76 | Depends(prog, graph_dependency) |
Michalis Spyrou | 165308c | 2019-04-10 10:10:21 +0100 | [diff] [blame] | 77 | prog = install_bin(prog) |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 78 | alias = examples_env.Alias(example, prog) |
| 79 | Default(alias) |
| 80 | |
Georgios Pinitas | 33893c3 | 2018-05-18 16:54:54 +0100 | [diff] [blame] | 81 | if env['opencl'] and env['neon']: |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 82 | for file in Glob("./neoncl_*.cpp"): |
| 83 | example = os.path.basename(os.path.splitext(str(file))[0]) |
Anthony Barbier | 7fb7b61 | 2018-05-16 11:58:52 +0100 | [diff] [blame] | 84 | prog = examples_env.Program(example, ["{}.cpp".format(example), utils], CPPDEFINES=['ARM_COMPUTE_CL'], LIBS = examples_libs + arm_compute_libs) |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 85 | Depends(prog, arm_compute_dependency) |
Michalis Spyrou | 165308c | 2019-04-10 10:10:21 +0100 | [diff] [blame] | 86 | prog = install_bin(prog) |
Anthony Barbier | 2a07e18 | 2017-08-04 18:20:27 +0100 | [diff] [blame] | 87 | alias = examples_env.Alias(example, prog) |
| 88 | Default(alias) |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 89 | |
| 90 | if env['opencl']: |
| 91 | for file in Glob("./cl_*.cpp"): |
| 92 | example = os.path.basename(os.path.splitext(str(file))[0]) |
Anthony Barbier | 7fb7b61 | 2018-05-16 11:58:52 +0100 | [diff] [blame] | 93 | prog = examples_env.Program(example, ["{}.cpp".format(example), utils], CPPDEFINES=['ARM_COMPUTE_CL'], LIBS = examples_libs + arm_compute_libs) |
Georgios Pinitas | d9eb275 | 2018-04-03 13:44:29 +0100 | [diff] [blame] | 94 | Depends(prog, arm_compute_dependency) |
Michalis Spyrou | 165308c | 2019-04-10 10:10:21 +0100 | [diff] [blame] | 95 | prog = install_bin(prog) |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 96 | alias = examples_env.Alias(example, prog) |
| 97 | Default(alias) |
| 98 | |
SiCong Li | 8b4c730 | 2019-09-19 12:18:15 +0100 | [diff] [blame] | 99 | if env['gemm_tuner'] and env['opencl']: |
SiCong Li | 240b79d | 2019-09-24 15:50:34 +0100 | [diff] [blame] | 100 | gemm_tuner_common_options = examples_env.Object("./gemm_tuner/CommonGemmExampleOptions.cpp") |
SiCong Li | 8b4c730 | 2019-09-19 12:18:15 +0100 | [diff] [blame] | 101 | for file in Glob("./gemm_tuner/cl_*.cpp"): |
| 102 | example = os.path.basename(os.path.splitext(str(file))[0]) |
| 103 | example = os.path.join("gemm_tuner", example) |
Georgios Pinitas | 4551403 | 2020-12-30 00:03:09 +0000 | [diff] [blame] | 104 | if env['os'] in ['android', 'macos', 'bare_metal'] or env['standalone']: |
| 105 | prog = examples_env.Program(example, ["{}.cpp".format(example), utils, gemm_tuner_common_options], CPPDEFINES=['ARM_COMPUTE_CL'], LIBS = examples_libs + arm_compute_graph_libs, LINKFLAGS=examples_env["LINKFLAGS"]+[load_whole_archive, graph_dependency, noload_whole_archive, '-fstack-protector-strong'] ) |
Eren Kopuz | 350099e | 2020-06-09 15:37:43 +0100 | [diff] [blame] | 106 | Depends(prog, graph_dependency) |
| 107 | prog = install_bin(prog) |
| 108 | else: |
| 109 | #-Wl,--allow-shlib-undefined: Ignore dependencies of dependencies |
| 110 | prog = examples_env.Program(example, ["{}.cpp".format(example), utils, gemm_tuner_common_options], CPPDEFINES=['ARM_COMPUTE_CL'], LIBS = examples_libs + arm_compute_graph_libs, LINKFLAGS=examples_env["LINKFLAGS"]+['-Wl,--allow-shlib-undefined'] ) |
| 111 | Depends(prog, graph_dependency) |
| 112 | prog = install_bin(prog) |
SiCong Li | 8b4c730 | 2019-09-19 12:18:15 +0100 | [diff] [blame] | 113 | alias = examples_env.Alias(example, prog) |
| 114 | Default(alias) |
| 115 | |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 116 | if env['neon']: |
| 117 | for file in Glob("./neon_*.cpp"): |
| 118 | example = os.path.basename(os.path.splitext(str(file))[0]) |
Michalis Spyrou | 0183368 | 2020-01-09 15:15:13 +0000 | [diff] [blame] | 119 | |
| 120 | prog = None |
| 121 | if env['os'] in ['bare_metal']: |
Michalis Spyrou | 6a9e801 | 2020-02-07 09:45:55 +0000 | [diff] [blame] | 122 | prog = examples_env.Program(example, ["{}.cpp".format(example), utils], LINKFLAGS=examples_env["LINKFLAGS"], LIBS = examples_libs + arm_compute_libs) |
Michalis Spyrou | 0183368 | 2020-01-09 15:15:13 +0000 | [diff] [blame] | 123 | else: |
| 124 | prog = examples_env.Program(example, ["{}.cpp".format(example), utils], LIBS = examples_libs + arm_compute_libs) |
| 125 | |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 126 | Depends(prog, arm_compute_dependency) |
Michalis Spyrou | 165308c | 2019-04-10 10:10:21 +0100 | [diff] [blame] | 127 | prog = install_bin(prog) |
Anthony Barbier | 6ff3b19 | 2017-09-04 18:44:23 +0100 | [diff] [blame] | 128 | alias = examples_env.Alias(example, prog) |
| 129 | Default(alias) |
Anthony Barbier | 7068f99 | 2017-10-26 15:23:08 +0100 | [diff] [blame] | 130 | |
Michele Di Giorgio | 72610dc | 2020-11-18 15:29:08 +0000 | [diff] [blame] | 131 | if env['external_tests_dir']: |
| 132 | for file in Glob(env['external_tests_dir'] + "/examples/graph_*.cpp"): |
| 133 | example = os.path.basename(os.path.splitext(str(file))[0]) |
| 134 | prog = None |
Georgios Pinitas | 33893c3 | 2018-05-18 16:54:54 +0100 | [diff] [blame] | 135 | |
Georgios Pinitas | 4551403 | 2020-12-30 00:03:09 +0000 | [diff] [blame] | 136 | if env['os'] in ['android', 'macos', 'bare_metal'] or env['standalone']: |
| 137 | prog = examples_env.Program(example, [examples_env.Object(source=file, target=example), utils, graph_utils], LIBS = examples_libs + arm_compute_graph_libs, LINKFLAGS=examples_env["LINKFLAGS"]+[load_whole_archive, graph_dependency, noload_whole_archive] + extra_link_flags) |
Michele Di Giorgio | 72610dc | 2020-11-18 15:29:08 +0000 | [diff] [blame] | 138 | Depends(prog, graph_dependency) |
| 139 | prog = install_bin(prog) |
| 140 | else: |
| 141 | #-Wl,--allow-shlib-undefined: Ignore dependencies of dependencies |
| 142 | prog = examples_env.Program(example, [examples_env.Object(source=file, target=example), utils, graph_utils], LIBS = examples_libs + arm_compute_graph_libs, LINKFLAGS=examples_env["LINKFLAGS"]+['-Wl,--allow-shlib-undefined'] ) |
| 143 | Depends(prog, graph_dependency) |
| 144 | prog = install_bin(prog) |
| 145 | alias = examples_env.Alias(example, prog) |
| 146 | Default(alias) |