summaryrefslogtreecommitdiff
path: root/cmake/Templates/caffe_config.h.in
blob: 9302022d7dad6a76cf2521a80718e6f230879c67 (plain)
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
/* Sources directory */
#define SOURCE_FOLDER "${PROJECT_SOURCE_DIR}"

/* Binaries directory */
#define BINARY_FOLDER "${PROJECT_BINARY_DIR}"

/* NVIDA Cuda */
#cmakedefine HAVE_CUDA

/* NVIDA cuDNN */
#cmakedefine HAVE_CUDNN
#cmakedefine USE_CUDNN

/* NVIDA cuDNN */
#cmakedefine CPU_ONLY

/* Test device */
#define CUDA_TEST_DEVICE ${CUDA_TEST_DEVICE}

/* Temporary (TODO: remove) */
#if 1
  #define CMAKE_SOURCE_DIR SOURCE_FOLDER "/src/"
  #define EXAMPLES_SOURCE_DIR BINARY_FOLDER "/examples/"
  #define CMAKE_EXT ".gen.cmake"
#else
  #define CMAKE_SOURCE_DIR "src/"
  #define EXAMPLES_SOURCE_DIR "examples/"
  #define CMAKE_EXT ""
#endif

/* Matlab */
#cmakedefine HAVE_MATLAB

/* IO libraries */
#cmakedefine USE_OPENCV
#cmakedefine USE_LMDB
#cmakedefine USE_LEVELDB