blob: e363af2ae9b2b56bc730c06c12a69d0ea772a83b [file] [log] [blame]
#-------------------------------------------------------------------------------
# Template configuration for compiling nnvm
#
# If you want to change the configuration, please use the following
# steps. Assume you are on the root directory of nnvm. First copy the this
# file so that any local changes will be ignored by git
#
# $ cp make/config.mk .
#
# Next modify the according entries, and then compile by
#
# $ make
#
# or build in parallel with 8 threads
#
# $ make -j8
#-------------------------------------------------------------------------------
#---------------------
# choice of compiler
#--------------------
export NVCC = nvcc
# choice of archiver
export AR = ar
# the additional link flags you want to add
ADD_LDFLAGS=
# the additional compile flags you want to add
ADD_CFLAGS=
# path to dmlc-core module
#DMLC_CORE_PATH=
#----------------------------
# plugins
#----------------------------
# whether to use fusion integration. This requires installing cuda.
# ifndef CUDA_PATH
# CUDA_PATH = /usr/local/cuda
# endif
# NNVM_FUSION_PATH = plugin/nnvm-fusion
# NNVM_PLUGINS += $(NNVM_FUSION_PATH)/nnvm-fusion.mk