forked from huawei/mindspore2022
21 lines
650 B
CMake
21 lines
650 B
CMake
add_subdirectory(datasetops)
|
|
if (ENABLE_TDTQUE)
|
|
add_subdirectory(tdt)
|
|
endif ()
|
|
|
|
file(GLOB_RECURSE _CURRENT_SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.cc")
|
|
set_property(SOURCE ${_CURRENT_SRC_FILES} PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_MD)
|
|
add_library(engine OBJECT
|
|
execution_tree.cc
|
|
data_buffer.cc
|
|
data_schema.cc
|
|
dataset_iterator.cc
|
|
)
|
|
target_include_directories(engine PRIVATE ${pybind11_INCLUDE_DIRS})
|
|
|
|
if (ENABLE_TDTQUE)
|
|
add_dependencies(engine engine-datasetops engine-datasetops-source engine-tdt)
|
|
else()
|
|
add_dependencies(engine engine-datasetops engine-datasetops-source)
|
|
endif ()
|