diff --git a/mindspore/lite/minddata/CMakeLists.txt b/mindspore/lite/minddata/CMakeLists.txt index dbc0d1c158..fa274c6fd9 100644 --- a/mindspore/lite/minddata/CMakeLists.txt +++ b/mindspore/lite/minddata/CMakeLists.txt @@ -51,6 +51,8 @@ AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/datasetops MINDDATA_ENGINE_DATASETOP 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) AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/datasetops/source/sampler MINDDATA_ENGINE_DATASETOPS_SOURCE_SAMPLER_SRC_FILES) +AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/ir/datasetops MINDDATA_ENGINE_IR_DATASETOPS_SRC_FILES) +AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/ir/datasetops/source MINDDATA_ENGINE_IR_DATASETOPS_SOURCE_SRC_FILES) AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/opt MINDDATA_ENGINE_OPT_SRC_FILES) AUX_SOURCE_DIRECTORY(${MINDDATA_DIR}/engine/opt/optional MINDDATA_ENGINE_OPT_OPTIONAL_SRC_FILES) @@ -143,6 +145,8 @@ if (BUILD_MINDDATA STREQUAL "full") ${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_OPT_SRC_FILES} ${MINDDATA_ENGINE_OPT_OPTIONAL_SRC_FILES} ${MINDDATA_ENGINE_OPT_POST_SRC_FILES}