2020-11-13 09:53:23 +00:00
|
|
|
# Copyright (C) 2019 Intel Corporation. All rights reserved.
|
|
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
|
|
|
|
cmake_minimum_required (VERSION 3.0)
|
|
|
|
|
|
|
|
project(xnnpack_wasm)
|
|
|
|
|
|
|
|
################ EMCC ################
|
|
|
|
if(NOT DEFINED ENV{EMSDK})
|
|
|
|
message(SEND_ERROR
|
|
|
|
"can not find emsdk. "
|
|
|
|
"please refer to https://emscripten.org/docs/getting_started/downloads.html "
|
|
|
|
"and install it, "
|
|
|
|
"or active emsdk by 'source ./emsdk_env.sh'"
|
|
|
|
)
|
|
|
|
endif()
|
|
|
|
|
|
|
|
include(ExternalProject)
|
|
|
|
|
|
|
|
ExternalProject_Add(xnnpack
|
|
|
|
PREFIX xnnpack
|
|
|
|
GIT_REPOSITORY https://github.com/google/XNNPACK.git
|
2021-04-27 09:05:40 +00:00
|
|
|
GIT_TAG master
|
2020-11-13 09:53:23 +00:00
|
|
|
GIT_PROGRESS ON
|
|
|
|
SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/xnnpack
|
2021-04-27 09:05:40 +00:00
|
|
|
UPDATE_COMMAND git checkout .
|
2020-11-13 09:53:23 +00:00
|
|
|
&& git apply ${CMAKE_CURRENT_SOURCE_DIR}/xnnpack.patch
|
|
|
|
CONFIGURE_COMMAND ""
|
|
|
|
BUILD_COMMAND cd ${CMAKE_CURRENT_SOURCE_DIR}/xnnpack
|
2021-04-27 09:05:40 +00:00
|
|
|
&& bazel --output_user_root=build_user_output build -c opt --config=wasm
|
|
|
|
//:qs8_gemm_bench
|
|
|
|
//:qs8_requantization_bench
|
|
|
|
//:qu8_gemm_bench
|
|
|
|
//:qu8_requantization_bench
|
|
|
|
//:f16_igemm_bench
|
|
|
|
//:f16_gemm_bench
|
|
|
|
//:f16_spmm_bench
|
|
|
|
//:f32_igemm_bench
|
|
|
|
//:f16_relu_bench
|
|
|
|
//:f32_conv_hwc_bench
|
|
|
|
//:f32_conv_hwc2chw_bench
|
|
|
|
//:f16_dwconv_bench
|
|
|
|
//:f32_dwconv_bench
|
|
|
|
//:f32_dwconv2d_chw_bench
|
|
|
|
//:f32_gemm_bench
|
|
|
|
//:f32_hswish_bench
|
|
|
|
//:f32_raddexpminusmax_bench
|
|
|
|
//:f32_raddextexp_bench
|
|
|
|
//:f32_raddstoreexpminusmax_bench
|
|
|
|
//:f32_relu_bench
|
|
|
|
//:f32_rmax_bench
|
|
|
|
//:f32_sigmoid_bench
|
|
|
|
//:f32_spmm_bench
|
|
|
|
//:f32_softmax_bench
|
|
|
|
//:f32_velu_bench
|
|
|
|
//:f32_vscaleexpminusmax_bench
|
|
|
|
//:f32_vscaleextexp_bench
|
|
|
|
//:f32_vsqrt_bench
|
|
|
|
//:f32_im2col_gemm_bench
|
|
|
|
//:rounding_bench
|
|
|
|
//:average_pooling_bench
|
|
|
|
//:bankers_rounding_bench
|
|
|
|
//:ceiling_bench
|
|
|
|
//:channel_shuffle_bench
|
|
|
|
//:convolution_bench
|
|
|
|
//:deconvolution_bench
|
|
|
|
//:elu_bench
|
|
|
|
//:floor_bench
|
|
|
|
//:global_average_pooling_bench
|
|
|
|
//:hardswish_bench
|
|
|
|
//:max_pooling_bench
|
|
|
|
//:sigmoid_bench
|
|
|
|
//:prelu_bench
|
|
|
|
//:softmax_bench
|
|
|
|
//:square_root_bench
|
|
|
|
//:truncation_bench
|
|
|
|
//:f32_dwconv_e2e_bench
|
|
|
|
//:f32_gemm_e2e_bench
|
|
|
|
//:qs8_gemm_e2e_bench
|
|
|
|
//:end2end_bench
|
|
|
|
//:f32_exp_ulp_eval
|
|
|
|
//:f32_expminus_ulp_eval
|
|
|
|
//:f32_expm1minus_ulp_eval
|
|
|
|
//:f32_extexp_ulp_eval
|
|
|
|
//:f32_sigmoid_ulp_eval
|
|
|
|
//:f32_sqrt_ulp_eval
|
2020-11-13 09:53:23 +00:00
|
|
|
#--sandbox_debug
|
2021-04-27 09:05:40 +00:00
|
|
|
INSTALL_COMMAND ${CMAKE_COMMAND} -E copy_directory
|
2020-11-13 09:53:23 +00:00
|
|
|
${CMAKE_CURRENT_SOURCE_DIR}/xnnpack/bazel-out/wasm-opt/bin/
|
|
|
|
${CMAKE_CURRENT_SOURCE_DIR}/build/wasm-opt
|
|
|
|
)
|