CMakeLists.txt

cmake_minimum_required(VERSION 3.13)

set(SNITCH_SOFTWARE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/snitch/sw/)
list(APPEND CMAKE_MODULE_PATH ${SNITCH_SOFTWARE_DIR}/cmake)
set(CMAKE_TOOLCHAIN_FILE toolchain-llvm CACHE STRING "Toolchain to use")

project(snitch-onnx LANGUAGES C ASM)
if (CLUSTER_SIM)
    message("Linking against snRuntime-cluster. Use snitch_cluster.vlt to simulate")
    set(SNITCH_SIMULATOR ${CMAKE_CURRENT_SOURCE_DIR}/snitch/hw/system/snitch_cluster/bin/snitch_cluster.vlt CACHE PATH "")
    set(TARGET_LINKER_SCRIPT ${CMAKE_CURRENT_SOURCE_DIR}/snitch/hw/system/snitch_cluster/generated/link.ld CACHE PATH "")
    set(SNITCH_RUNTIME snRuntime-cluster CACHE STRING "")
else()
    message("Using banshee. Use banshee BINARY to simulate. (f.ex: --configuration $SNITCH_ROOT/sw/banshee/config/snitch_cluster.yaml -l")
    set(SNITCH_RUNTIME snRuntime-banshee CACHE STRING "")
endif()

include(SnitchUtilities)

add_compile_options(-O3 -g -ffunction-sections)

# Build the runtime
add_subdirectory(${SNITCH_SOFTWARE_DIR}snRuntime snRuntime)
add_subdirectory(${SNITCH_SOFTWARE_DIR}benchmark benchmark)


include_directories(${SNRUNTIME_INCLUDE_DIRS})

# LMQ includes
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src/lmq)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src/onnx)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src/dot)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src/benchmark)

# Define size for the benchmark
if (LMQ_SIZE)
    message("Setting size to ${LMQ_SIZE}")
    add_compile_definitions(LMQ_SIZE=${LMQ_SIZE})
else()
    message("no size defined")
endif()

add_snitch_executable(hello_world
                      ./src/lmq/lmq.c
                      ./src/hello_world/main.c)
add_snitch_executable(debug-multicore ./src/debug-multicore/main.c)

add_snitch_executable(ssr_anomaly
                      ./src/lmq/lmq.c
                      ./src/bugs/ssr_anomaly.c)

# x86 gemm
# execute_process is needed as cmake does not support different compilers within one project.
execute_process(
    COMMAND bash "-c" "gcc ${CMAKE_CURRENT_SOURCE_DIR}/src/x86/main.c -O3 -Wall -Wextra -lm -fopenmp -o x86_gemm" 
)


# Compile 'sum' library and its corresponding benchmark
add_library(summation src/onnx/sum.c)
add_snitch_executable(benchmark_sum
                      ./src/benchmark/benchmark_sum.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_sum summation)

# Compile 'add'
add_library(add src/onnx/add.c)
add_snitch_executable(benchmark_add
                      ./src/benchmark/benchmark_add.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_add add)

# Compile 'div'
add_library(div src/onnx/div.c)
add_snitch_executable(benchmark_div
                      ./src/benchmark/benchmark_div.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_div div)

# Compile 'abs'
add_library(abs src/onnx/abs.c)
add_snitch_executable(benchmark_abs
                      ./src/benchmark/benchmark_abs.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_abs abs)

#Compile 'copy'
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src/copy)
add_library(copy src/copy/copy.c)
add_snitch_executable(benchmark_copy
                      ./src/benchmark/benchmark_copy.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_copy copy)

# Compile 'relu'
add_library(relu src/onnx/relu.c)
add_snitch_executable(benchmark_relu
                      ./src/lmq/lmq.c
                      ./src/benchmark/benchmark_relu.c)
target_link_libraries(benchmark_relu relu)

# Compile 'acos'
add_library(acos src/onnx/acos.c)
add_snitch_executable(benchmark_acos
                      ./src/benchmark/benchmark_acos.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_acos acos)

# Compile 'acosh'
add_library(acosh src/onnx/acosh.c)
add_snitch_executable(benchmark_acosh
                      ./src/benchmark/benchmark_acosh.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_acosh acosh)

# Compile 'asinh'
add_library(asinh src/onnx/asinh.c)
add_snitch_executable(benchmark_asinh
                      ./src/benchmark/benchmark_asinh.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_asinh asinh)


# Compile 'sigmoid'
add_library(sigmoid src/onnx/sigmoid.c)
add_snitch_executable(benchmark_sigmoid
                      ./src/benchmark/benchmark_sigmoid.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_sigmoid sigmoid)

# Compile 'argmax'
add_library(argmax src/onnx/argmax.c)
add_snitch_executable(benchmark_argmax
                      ./src/benchmark/benchmark_argmax.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_argmax argmax)

# Compile 'max'
add_library(max src/onnx/max.c)
add_snitch_executable(benchmark_max
                      ./src/benchmark/benchmark_max.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_max max)

# Compile 'batchnorm'
add_library(batchnorm src/onnx/batchnorm.c)
add_snitch_executable(benchmark_batchnorm
                      ./src/benchmark/benchmark_batchnorm.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_batchnorm batchnorm)

# Compile 'cumsum'
add_library(cumsum src/onnx/cumsum.c)
add_snitch_executable(benchmark_cumsum
                      ./src/benchmark/benchmark_cumsum.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_cumsum cumsum)

# Compile 'dropout'
add_library(dropout src/onnx/dropout.c)
add_snitch_executable(benchmark_dropout
                      ./src/benchmark/benchmark_dropout.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_dropout dropout)

# Compile 'transpose'
add_library(transpose src/onnx/transpose.c)
add_snitch_executable(benchmark_transpose
                      ./src/benchmark/benchmark_transpose.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_transpose transpose)

# Compile 'masked_dropout'
add_library(masked_dropout src/onnx/masked_dropout.c)
add_snitch_executable(benchmark_masked_dropout
                      ./src/benchmark/benchmark_masked_dropout.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_masked_dropout masked_dropout)

# Compile 'dot'
add_library(dot src/dot/dot.c)
add_snitch_executable(benchmark_dot
                      ./src/benchmark/benchmark_dot.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_dot dot)

# Compile memory test
add_snitch_executable(benchmark_mem_versus_l1
                      ./src/benchmark/benchmark_mem_versus_l1.c
                      ./src/lmq/lmq.c)
# Compile 'sin'
add_library(sin src/onnx/sin.c)
add_snitch_executable(benchmark_sin
                      ./src/benchmark/benchmark_sin.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_sin sin)

# Compile 'maxpool' and 'maxpool2d'
add_library(maxpool src/onnx/maxpool.c)
add_snitch_executable(benchmark_maxpool ./src/benchmark/benchmark_maxpool.c ./src/lmq/lmq.c)
target_link_libraries(benchmark_maxpool maxpool)
add_snitch_executable(benchmark_maxpool2d ./src/benchmark/benchmark_maxpool2d.c ./src/lmq/lmq.c)
target_link_libraries(benchmark_maxpool2d maxpool)

# Compile 'gemm'
add_library(gemm src/onnx/gemm.c)
add_snitch_executable(benchmark_gemm ./src/benchmark/benchmark_gemm.c ./src/lmq/lmq.c)
target_link_libraries(benchmark_gemm gemm)

# Compile 'conv' and 'conv2d'
add_library(conv src/onnx/conv.c)
add_snitch_executable(benchmark_conv ./src/benchmark/benchmark_conv.c ./src/lmq/lmq.c)
target_link_libraries(benchmark_conv conv)
add_snitch_executable(benchmark_conv2d ./src/benchmark/benchmark_conv2d.c ./src/lmq/lmq.c)
target_link_libraries(benchmark_conv2d conv)

# Compile 'unique'
add_library(unique src/onnx/unique.c)
add_snitch_executable(benchmark_unique
                      ./src/benchmark/benchmark_unique.c
                      ./src/lmq/lmq.c)
target_link_libraries(benchmark_unique unique)