blob: 19f4fe1fe9ed5dae1b37096e2286eb40b69bb7ff [file] [log] [blame]
cmake_minimum_required(VERSION 3.13.4)
# ------------------------------------------------------------------------------
# Project configuration
# ------------------------------------------------------------------------------
project(ml-models-public)
set_property(GLOBAL PROPERTY USE_FOLDERS ON)
set(SPARROW_IREE_DIR "$ENV{ROOTDIR}/sw/vec_iree" CACHE PATH
"Sparrow IREE library path (default: $ENV{ROOTDIR}/sw/vec_iree)")
include_directories(BEFORE SYSTEM ${CMAKE_CURRENT_LIST_DIR})
include_directories(BEFORE SYSTEM ${CMAKE_CURRENT_BINARY_DIR})
include_directories(BEFORE SYSTEM ${SPARROW_IREE_DIR})
# Include sparrow_iree libraries.
add_subdirectory(${SPARROW_IREE_DIR} sparrow_iree)
# Add compilation options.
include(${SPARROW_IREE_DIR}/cmake/sparrow_ops.cmake)
# Set default linker memory size for the whole project.
set(TCM_LENGTH "16M" CACHE STRING "TCM Length (default: 16M)")
add_link_options("LINKER:--defsym=__tcm_length__=${TCM_LENGTH}")
# Include model directories.
add_subdirectory(float_models)
add_subdirectory(quant_models)