39 #ifndef MXNET_USE_OPENCV 40 #define MXNET_USE_OPENCV 0 46 #ifndef MXNET_USE_CUDA 47 #define MXNET_USE_CUDA MSHADOW_USE_CUDA 53 #ifndef MXNET_USE_CUDNN 54 #define MXNET_USE_CUDNN MSHADOW_USE_CUDNN 57 #ifndef MXNET_USE_NCCL 58 #define MXNET_USE_NCCL 0 64 #ifndef MXNET_USE_CUSOLVER 65 #define MXNET_USE_CUSOLVER MSHADOW_USE_CUSOLVER 68 #ifndef MXNET_ENABLE_CUDA_RTC 69 #define MXNET_ENABLE_CUDA_RTC 0 73 #define MXNET_GPU_NOT_ENABLED_ERROR "GPU is not enabled" 76 #ifndef MXNET_USE_TENSORRT 77 #define MXNET_USE_TENSORRT 0 81 #ifndef MXNET_USE_BLAS_ATLAS 82 #define MXNET_USE_BLAS_ATLAS 0 85 #ifndef MXNET_USE_BLAS_OPEN 86 #define MXNET_USE_BLAS_OPEN 0 89 #ifndef MXNET_USE_BLAS_MKL 90 #define MXNET_USE_BLAS_MKL 0 93 #ifndef MXNET_USE_BLAS_APPLE 94 #define MXNET_USE_BLAS_APPLE 0 97 #ifndef MXNET_USE_LAPACK 98 #define MXNET_USE_LAPACK 0 101 #ifndef MXNET_USE_MKLDNN 102 #define MXNET_USE_MKLDNN 0 105 #ifndef MXNET_USE_OPENMP 106 #define MXNET_USE_OPENMP 0 109 #ifndef MXNET_USE_F16C 110 #define MXNET_USE_F16C MSHADOW_USE_F16C 113 #ifndef MXNET_USE_CAFFE 114 #define MXNET_USE_CAFFE 0 117 #ifndef MXNET_USE_DIST_KVSTORE 118 #define MXNET_USE_DIST_KVSTORE 0 121 #ifndef MXNET_USE_SIGNAL_HANDLER 122 #define MXNET_USE_SIGNAL_HANDLER 0 125 #ifndef MXNET_USE_INT64_TENSOR_SIZE 126 #define MXNET_USE_INT64_TENSOR_SIZE MSHADOW_INT64_TENSOR_SIZE 129 #ifndef MXNET_USE_TVM_OP 130 #define MXNET_USE_TVM_OP 0 200 static const std::vector<std::string>
names;
207 return m_lib_features;
210 std::array<LibFeature, MAX_FEATURES> m_lib_features;
211 static std::unique_ptr<LibInfo> m_inst;
Definition: libinfo.h:162
Definition: libinfo.h:189
Definition: libinfo.h:155
Definition: libinfo.h:161
Definition: libinfo.h:166
Definition: libinfo.h:153
namespace of mxnet
Definition: api_registry.h:33
Definition: libinfo.h:143
Definition: libinfo.h:173
Definition: libinfo.h:152
Definition: libinfo.h:142
Definition: libinfo.h:185
Definition: libinfo.h:141
Definition: libinfo.h:170
Definition: libinfo.h:178
Definition: libinfo.h:145
Definition: libinfo.h:150
Definition: libinfo.h:167
bool is_enabled(unsigned feat)
Definition: libinfo.h:184
Definition: libinfo.h:188
Definition: libinfo.h:160
Definition: libinfo.h:148
Definition: libinfo.h:183
const std::array< LibFeature, MAX_FEATURES > & getFeatures()
Definition: libinfo.h:206
Definition: libinfo.h:144
Definition: libinfo.h:203
Definition: libinfo.h:182
Definition: libinfo.h:169
Definition: libinfo.h:151
Definition: libinfo.h:159
Definition: libinfo.h:199
Definition: libinfo.h:195
Definition: libinfo.h:154
Definition: libinfo.h:181
Definition: libinfo.h:175
Definition: libinfo.h:149
static const std::vector< std::string > names
Definition: libinfo.h:200
Definition: libinfo.h:192