Browse Source

!7597 Add missing folder to Minddata

Merge pull request !7597 from EricZ/add_ir_folder
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
ed4bae6a3e
1 changed files with 14 additions and 4 deletions
  1. +14
    -4
      mindspore/lite/minddata/CMakeLists.txt

+ 14
- 4
mindspore/lite/minddata/CMakeLists.txt View File

@@ -47,6 +47,7 @@ AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine MINDDATA_ENGINE_SRC_FILES)
# TODO: add cache
AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/cache MINDDATA_ENGINE_CACHE_SRC_FILES)

AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/consumers MINDDATA_ENGINE_CONSUMERS_SRC_FILES)
AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/datasetops MINDDATA_ENGINE_DATASETOPS_SRC_FILES)
AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/datasetops/map_op MINDDATA_ENGINE_DATASETOPS_MAPOP_SRC_FILES)
AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/datasetops/source MINDDATA_ENGINE_DATASETOPS_SOURCE_SRC_FILES)
@@ -114,7 +115,7 @@ if (BUILD_MINDDATA STREQUAL "full")
"${MINDDATA_DIR}/engine/datasetops/source/sampler/python_sampler.cc"
)

list(REMOVE_ITEM MINDDATA_ENGINE_OPT_POST_SRC_FILES
list(REMOVE_ITEM MINDDATA_ENGINE_OPT_POST_SRC_FILES
"${MINDDATA_DIR}/engine/opt/post/repeat_pass.cc"
)

@@ -127,7 +128,15 @@ if (BUILD_MINDDATA STREQUAL "full")
#"${MINDDATA_DIR}/kernels/image/affine_op.cc"
#"${MINDDATA_DIR}/kernels/image/cutmix_batch_op.cc"
#"${MINDDATA_DIR}/kernels/image/random_affine_op.cc"
#)

list(REMOVE_ITEM MINDDATA_ENGINE_IR_DATASETOPS_SOURCE_SRC_FILES
"${MINDDATA_DIR}/engine/ir/datasetops/source/generator_node.cc"
)

list(REMOVE_ITEM MINDDATA_ENGINE_IR_DATASETOPS_SRC_FILES
"${MINDDATA_DIR}/engine/ir/datasetops/bucket_batch_by_length_node.cc"
"${MINDDATA_DIR}/engine/ir/datasetops/build_vocab_node.cc"
)

include_directories("${CMAKE_BINARY_DIR}/minddata/dataset/engine/cache")

@@ -141,12 +150,13 @@ if (BUILD_MINDDATA STREQUAL "full")
${MINDDATA_CORE_SRC_FILES}
${MINDDATA_ENGINE_SRC_FILES}
#${MINDDATA_ENGINE_CACHE_SRC_FILES}
${MINDDATA_ENGINE_CONSUMERS_SRC_FILES}
${MINDDATA_ENGINE_DATASETOPS_SRC_FILES}
${MINDDATA_ENGINE_DATASETOPS_MAPOP_SRC_FILES}
${MINDDATA_ENGINE_DATASETOPS_SOURCE_SRC_FILES}
${MINDDATA_ENGINE_DATASETOPS_SOURCE_SAMPLER_SRC_FILES}
${MINDDATA_ENGINE_IR_DATASETOPS_SRC_FILES}
${MINDDATA_ENGINE_IR_DATASETOPS_SOURCE_SRC_FILES}
${MINDDATA_ENGINE_IR_DATASETOPS_SRC_FILES}
${MINDDATA_ENGINE_IR_DATASETOPS_SOURCE_SRC_FILES}
${MINDDATA_ENGINE_OPT_SRC_FILES}
${MINDDATA_ENGINE_OPT_OPTIONAL_SRC_FILES}
${MINDDATA_ENGINE_OPT_POST_SRC_FILES}


Loading…
Cancel
Save