Browse Source

add lstm cmakelist

pull/1498/head
dongduo5@huawei.com 4 years ago
parent
commit
a1c316ffdf
5 changed files with 6 additions and 0 deletions
  1. +2
    -0
      ge/CMakeLists.txt
  2. +1
    -0
      ge/common/CMakeLists.txt
  3. +1
    -0
      ge/ge_inference.mk
  4. +1
    -0
      ge/ge_runner.mk
  5. +1
    -0
      tests/ut/ge/CMakeLists.txt

+ 2
- 0
ge/CMakeLists.txt View File

@@ -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"


+ 1
- 0
ge/common/CMakeLists.txt View File

@@ -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"


+ 1
- 0
ge/ge_inference.mk View File

@@ -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 \


+ 1
- 0
ge/ge_runner.mk View File

@@ -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 \


+ 1
- 0
tests/ut/ge/CMakeLists.txt View File

@@ -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"


Loading…
Cancel
Save