cmake_minimum_required(VERSION 3.5.1)
project(mininn)

set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

# for RISC-V porting
set(CMAKE_SYSTEM_NAME Generic)
set(CMAKE_SYSTEM_PROCESSOR riscv)
set(CMAKE_C_COMPILER riscv64-unknown-linux-gnu-gcc)
set(CMAKE_CXX_COMPILER riscv64-unknown-linux-gnu-g++)

# segment fault when linking statically
# set(CMAKE_FIND_LIBRARY_SUFFIXES ".a")
# set(BUILD_SHARED_LIBS OFF)
# set(CMAKE_EXE_LINKER_FLAGS "-static")

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -O3 -fopenmp")

set(APP_INCLUDES
        include/
        include/minidnn
 )

#Include the directories and now your cpp files will recognize your headers
include_directories(${APP_INCLUDES})

add_executable(mininn main.cpp)
target_sources(mininn PUBLIC mnist.cc)
TARGET_LINK_LIBRARIES(mininn
        pthread
        #"${TORCH_LIBRARIES}"
        )