Included more directories

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}