config.mk 1.96 KB
Newer Older
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
#-------------------------------------------------------------------------------
#  Template configuration for compiling
#
#  If you want to change the configuration, please use the following
#  steps. Assume you are on the root directory. 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
#-------------------------------------------------------------------------------

# whether compile with debug
DEBUG = 0

# the additional link flags you want to add
ADD_LDFLAGS =

# the additional compile flags you want to add
ADD_CFLAGS =

#---------------------------------------------
29
# Backend runtimes.
30
#---------------------------------------------
31
# whether enable CUDA during compile
32
USE_CUDA = 0
33

34 35 36 37
# add the path to CUDA library to link and compile flag
# if you have already add them to environment variable.
# CUDA_PATH = /usr/local/cuda

38 39 40
# ROCM
USE_ROCM = 0

41
# whether enable OpenCL during compile
42
USE_OPENCL = 0
43 44

# whether enable Metal during compile
45
USE_METAL = 0
46

47
# Whether enable RPC during compile
48 49 50 51
USE_RPC = 1

# Whether enable tiny embedded graph runtime.
USE_GRAPH_RUNTIME = 1
52

53 54 55
#  Whether enable additional graph debug functions
USE_GRAPH_RUNTIME_DEBUG = 0

56 57
# whether build with LLVM support
# Requires LLVM version >= 4.0
58 59 60
# Set LLVM_CONFIG to your version, uncomment to build with llvm support
#
# LLVM_CONFIG = llvm-config
61 62 63 64 65 66

#---------------------------------------------
# Contrib optional libraries.
#---------------------------------------------
# Whether use BLAS, choices: openblas, atlas, blas, apple
USE_BLAS = none
67

68 69 70
# Whether use contrib.random in runtime
USE_RANDOM = 0

71
# Whether use NNPack
72 73 74
USE_NNPACK = 0
# NNPACK_PATH = none

75 76
# Whether use CuDNN
USE_CUDNN = 0
77

78 79 80
# Whether use MIOpen
USE_MIOPEN = 0

81 82
# Whether use MPS
USE_MPS = 0
83 84 85

# Whether use cuBLAS
USE_CUBLAS = 0
86 87 88

# Whether use rocBlas
USE_ROCBLAS = 0