From a1c316ffdfe04a405d7409a06b3ae6bdd3d146c5 Mon Sep 17 00:00:00 2001 From: "dongduo5@huawei.com" Date: Mon, 12 Apr 2021 15:04:23 +0800 Subject: [PATCH] add lstm cmakelist --- ge/CMakeLists.txt | 2 ++ ge/common/CMakeLists.txt | 1 + ge/ge_inference.mk | 1 + ge/ge_runner.mk | 1 + tests/ut/ge/CMakeLists.txt | 1 + 5 files changed, 6 insertions(+) diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt index d84bb89a..c3f79b6e 100755 --- a/ge/CMakeLists.txt +++ b/ge/CMakeLists.txt @@ -100,6 +100,7 @@ set(TRAIN_SRC_LIST "common/formats/format_transfers/format_transfer_nchw_nc1hwc0.cc" "common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.cc" "common/formats/format_transfers/format_transfer_transpose.cc" + "common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc" "common/formats/formats.cc" "common/formats/utils/formats_trans_utils.cc" "common/fp16_t.cc" @@ -434,6 +435,7 @@ set(INFER_SRC_LIST "common/formats/format_transfers/format_transfer_dhwcn_fracz3D.cc" "common/formats/format_transfers/format_transfer_dhwnc_fracz3D_transpose.cc" "common/formats/format_transfers/format_transfer_nchw_fz_c04.cc" + "common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc" "common/formats/formats.cc" "common/profiling/profiling_manager.cc" "common/dump/dump_properties.cc" diff --git a/ge/common/CMakeLists.txt b/ge/common/CMakeLists.txt index 75cb8ad1..51ac0e8a 100755 --- a/ge/common/CMakeLists.txt +++ b/ge/common/CMakeLists.txt @@ -49,6 +49,7 @@ set(SRC_LIST "formats/format_transfers/format_transfer_dhwcn_fracz3D.cc" "formats/format_transfers/format_transfer_dhwnc_fracz3D_transpose.cc" "formats/format_transfers/format_transfer_nchw_fz_c04.cc" + "formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc" "formats/formats.cc" "ge_format_util.cc" "fmk_error_codes.cc" diff --git a/ge/ge_inference.mk b/ge/ge_inference.mk index 32fc206d..8bbb0a6d 100755 --- a/ge/ge_inference.mk +++ b/ge/ge_inference.mk @@ -24,6 +24,7 @@ COMMON_LOCAL_SRC_FILES := \ common/formats/format_transfers/format_transfer_dhwcn_fracz3D.cc \ common/formats/format_transfers/format_transfer_dhwnc_fracz3D_transpose.cc \ common/formats/format_transfers/format_transfer_nchw_fz_c04.cc \ + common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc \ common/formats/formats.cc \ common/profiling/profiling_manager.cc \ common/dump/dump_properties.cc \ diff --git a/ge/ge_runner.mk b/ge/ge_runner.mk index 49515fe4..302b9c13 100644 --- a/ge/ge_runner.mk +++ b/ge/ge_runner.mk @@ -19,6 +19,7 @@ LIBGE_LOCAL_SRC_FILES := \ common/formats/format_transfers/format_transfer_nchw_nc1hwc0.cc \ common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.cc \ common/formats/format_transfers/format_transfer_transpose.cc \ + common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc \ common/formats/formats.cc \ common/formats/utils/formats_trans_utils.cc \ common/fp16_t.cc \ diff --git a/tests/ut/ge/CMakeLists.txt b/tests/ut/ge/CMakeLists.txt index 75985e4c..d1c5ed50 100755 --- a/tests/ut/ge/CMakeLists.txt +++ b/tests/ut/ge/CMakeLists.txt @@ -363,6 +363,7 @@ set(COMMON_FORMAT_SRC_FILES "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_nchw.cc" "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_nhwc.cc" "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_fracz_hwcn.cc" + "${GE_CODE_DIR}/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc \ "${GE_CODE_DIR}/ge/common/formats/utils/formats_trans_utils.cc" "${GE_CODE_DIR}/ge/graph/manager/util/hcom_util.cc" "${GE_CODE_DIR}/ge/common/dump/dump_manager.cc"