forked from dmlc/mshadow
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.mk
40 lines (34 loc) · 1.19 KB
/
config.mk
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
#---------------------------------------------------------------------------------------
# mshadow: the configuration compile script
#
# This is configuration script that you can use to compile mshadow
# Usage:
#
# include config.mk in your Makefile, or directly include the definition of variables
# include mshadow.mk after the variables are set
#
# Add MSHADOW_CFLAGS to the compile flags
# Add MSHADOW_LDFLAGS to the linker flags
# Add MSHADOW_NVCCFLAGS to the nvcc compile flags
#----------------------------------------------------------------------------------------
# whether use CUDA during compile
USE_CUDA = 0
# add the path to CUDA libary to link and compile flag
# if you have already add them to enviroment variable, leave it as NONE
USE_CUDA_PATH = NONE
#
# choose the version of blas you want to use
# can be: mkl, blas, atlas, openblas, apple
USE_BLAS = blas
#
# add path to intel library, you may need it
# for MKL, if you did not add the path to enviroment variable
#
USE_INTEL_PATH = NONE
# whether compile with parameter server
USE_DIST_PS = 0
PS_PATH = ../../../parameter_server/
PS_THIRD_PATH = NONE
# whether compile with rabit allreduce
USE_RABIT_PS = 1
RABIT_PATH = ../../rabit