diff --git a/mindspore/lite/micro/CMakeLists.txt b/mindspore/lite/micro/CMakeLists.txt index a84a331107..3471494b1d 100644 --- a/mindspore/lite/micro/CMakeLists.txt +++ b/mindspore/lite/micro/CMakeLists.txt @@ -16,12 +16,7 @@ if(NOT PLATFORM_ARM64 AND NOT PLATFORM_ARM32) include(${TOP_DIR}/cmake/external_libs/glog.cmake) include(${TOP_DIR}/cmake/external_libs/flatbuffers.cmake) - set(FBS_FILES - ${CMAKE_CURRENT_SOURCE_DIR}/../schema/model.fbs - ${CMAKE_CURRENT_SOURCE_DIR}/../schema/ops.fbs - ${CMAKE_CURRENT_SOURCE_DIR}/../schema/model_v0.fbs - ${CMAKE_CURRENT_SOURCE_DIR}/../schema/ops_v0.fbs - ) + file(GLOB FBS_FILES ${CMAKE_CURRENT_SOURCE_DIR}/../schema/*.fbs) ms_build_flatbuffers_lite(FBS_FILES ${CMAKE_CURRENT_SOURCE_DIR}/../schema/ fbs_src