From 6e27b4a303273432a560d42c55f89e21a512854b Mon Sep 17 00:00:00 2001 From: Eric Date: Wed, 21 Oct 2020 19:19:59 -0400 Subject: [PATCH] Included more directories --- mindspore/lite/minddata/CMakeLists.txt | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) diff --git a/mindspore/lite/minddata/CMakeLists.txt b/mindspore/lite/minddata/CMakeLists.txt index fa274c6fd96..378e6ea3ab9 100644 --- a/mindspore/lite/minddata/CMakeLists.txt +++ b/mindspore/lite/minddata/CMakeLists.txt @@ -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}