@@ -49,13 +49,22 @@ prepare_artifacts_upload() {
4949}
5050
5151build_cmake_executor_runner () {
52+ local backend_string_select=" ${1:- } "
5253 echo " Building executor_runner"
5354 rm -rf ${CMAKE_OUTPUT_DIR}
54- cmake -DCMAKE_BUILD_TYPE=Debug \
55- -DEXECUTORCH_BUILD_KERNELS_OPTIMIZED=ON \
56- -DPYTHON_EXECUTABLE=" $PYTHON_EXECUTABLE " \
57- -B${CMAKE_OUTPUT_DIR} .
58-
55+ if [[ " $backend_string_select " == " XNNPACK" ]]; then
56+ echo " Backend $backend_string_select selected"
57+ cmake -DCMAKE_BUILD_TYPE=Debug \
58+ -DEXECUTORCH_BUILD_KERNELS_OPTIMIZED=ON \
59+ -DEXECUTORCH_BUILD_XNNPACK=ON \
60+ -DPYTHON_EXECUTABLE=" $PYTHON_EXECUTABLE " \
61+ -B${CMAKE_OUTPUT_DIR} .
62+ else
63+ cmake -DCMAKE_BUILD_TYPE=Debug \
64+ -DEXECUTORCH_BUILD_KERNELS_OPTIMIZED=ON \
65+ -DPYTHON_EXECUTABLE=" $PYTHON_EXECUTABLE " \
66+ -B${CMAKE_OUTPUT_DIR} .
67+ fi
5968 cmake --build ${CMAKE_OUTPUT_DIR} -j4 --config Debug
6069}
6170
@@ -111,19 +120,6 @@ test_model() {
111120 run_portable_executor_runner
112121}
113122
114- build_cmake_xnn_executor_runner () {
115- echo " Building xnn_executor_runner"
116-
117- (rm -rf ${CMAKE_OUTPUT_DIR} \
118- && mkdir ${CMAKE_OUTPUT_DIR} \
119- && cd ${CMAKE_OUTPUT_DIR} \
120- && retry cmake -DCMAKE_BUILD_TYPE=Release \
121- -DEXECUTORCH_BUILD_XNNPACK=ON \
122- -DPYTHON_EXECUTABLE=" $PYTHON_EXECUTABLE " ..)
123-
124- cmake --build ${CMAKE_OUTPUT_DIR} -j4
125- }
126-
127123test_model_with_xnnpack () {
128124 WITH_QUANTIZATION=$1
129125 WITH_DELEGATION=$2
@@ -148,12 +144,13 @@ test_model_with_xnnpack() {
148144
149145 # Run test model
150146 if [[ " ${BUILD_TOOL} " == " buck2" ]]; then
147+ # TODO eventually buck should also use consolidated executor runners
151148 buck2 run //examples/xnnpack:xnn_executor_runner -- --model_path " ${OUTPUT_MODEL_PATH} "
152149 elif [[ " ${BUILD_TOOL} " == " cmake" ]]; then
153- if [[ ! -f ${CMAKE_OUTPUT_DIR} /backends/xnnpack/xnn_executor_runner ]]; then
154- build_cmake_xnn_executor_runner
150+ if [[ ! -f ${CMAKE_OUTPUT_DIR} /executor_runner ]]; then
151+ build_cmake_executor_runner( " XNNPACK " )
155152 fi
156- ./${CMAKE_OUTPUT_DIR} /backends/xnnpack/xnn_executor_runner --model_path " ${OUTPUT_MODEL_PATH} "
153+ ./${CMAKE_OUTPUT_DIR} /executor_runner --model_path " ${OUTPUT_MODEL_PATH} "
157154 else
158155 echo " Invalid build tool ${BUILD_TOOL} . Only buck2 and cmake are supported atm"
159156 exit 1
0 commit comments