arrayfire-ml/CMakeLists.txt at eea9572aeda3ba6e3452f35af914fb29df4c2bff · arrayfire/arrayfire-ml · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
cmake_minimum_required(VERSION 3.5.1)

project(ArrayFireML
  VERSION 0.1.0
  LANGUAGES C CXX)

find_package(ArrayFire REQUIRED)

add_library(afml SHARED "")

target_sources(afml
  PRIVATE
  src/autograd/Functions.cpp
  src/autograd/Variable.cpp
  src/nn/Modules/Activations.cpp
  src/nn/Modules/Container.cpp
  src/nn/Modules/Linear.cpp
  src/nn/Modules/Loss.cpp
  src/nn/Modules/Module.cpp
  src/nn/Modules/Dropout.cpp
  src/nn/Init.cpp
  src/optim/Optimizers.cpp
  )

target_include_directories(afml
  PUBLIC
  ${CMAKE_CURRENT_SOURCE_DIR}/include)

target_link_libraries(afml PUBLIC ArrayFire::af)

set_target_properties(afml
  PROPERTIES
  VERSION "${ArrayFireML_VERSION}"
  SOVERSION "${ArrayFireML_VERSION_MAJOR}"
  CXX_STANDARD 11)

add_subdirectory(examples)

option(PACKAGE_TESTS "Build tests" ON)
if(PACKAGE_TESTS)
    enable_testing()
    add_subdirectory(tests)
endif()