forked from dmlc/cxxnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
global.h
35 lines (31 loc) · 807 Bytes
/
global.h
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
#ifndef CXXNET_GLOBAL_H_
/*!
* \file global.h
* \brief global configuration of cxxnet, this controls how cxxnet is compiled
* \author Tianqi Chen
*/
/*! \brief whether to adapt caffe layers */
#ifndef CXXNET_USE_CAFFE_ADAPTOR
#define CXXNET_USE_CAFFE_ADAPTOR 0
#endif
/*!
*\brief whether to use opencv support,
* without it, we will not be able to use load jpg image iterarators
*/
#ifndef CXXNET_USE_OPENCV
#define CXXNET_USE_OPENCV 1
#endif
/*!
*\brief whether to use cudnn library for convolution
*/
#ifndef CXXNET_USE_CUDNN
#define CXXNET_USE_CUDNN 0
#endif
/*! \brief namespace of cxxnet */
namespace cxxnet {
typedef mshadow::cpu cpu;
typedef mshadow::gpu gpu;
typedef mshadow::index_t index_t;
typedef mshadow::default_real_t real_t;
} // namespace cxxnet
#endif // CXXNET_GLOBAL_H_