config.mk

#-------------------------------------------------------------------------------
#  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