libinfo.h File Reference

get features of the MXNet library at runtime More...

#include <string>
#include <vector>
#include <array>
#include <memory>
#include "dmlc/base.h"
#include "mshadow/base.h"
#include "c_api.h"

Include dependency graph for libinfo.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	mxnet::features::EnumNames
struct	mxnet::features::LibInfo

Namespaces
	mxnet
	namespace of mxnet
	mxnet::features

Macros
#define	MXNET_USE_OPENCV   0
	whether to use opencv support More...
#define	MXNET_USE_CUDNN   MSHADOW_USE_CUDNN
	whether to use cuda support More...
#define	MXNET_USE_NCCL   0
#define	MXNET_USE_CUSOLVER   MSHADOW_USE_CUSOLVER
	whether to use cusolver library More...
#define	MXNET_ENABLE_CUDA_RTC   0
#define	MXNET_GPU_NOT_ENABLED_ERROR   "GPU is not enabled"
	Error message for using gpu when MXNET_USE_CUDA==0. More...
#define	MXNET_USE_TENSORRT   0
#define	MXNET_USE_BLAS_ATLAS   0
#define	MXNET_USE_BLAS_OPEN   0
#define	MXNET_USE_BLAS_MKL   0
#define	MXNET_USE_BLAS_APPLE   0
#define	MXNET_USE_LAPACK   0
#define	MXNET_USE_MKLDNN   0
#define	MXNET_USE_OPENMP   0
#define	MXNET_USE_F16C   MSHADOW_USE_F16C
#define	MXNET_USE_CAFFE   0
#define	MXNET_USE_DIST_KVSTORE   0
#define	MXNET_USE_SIGNAL_HANDLER   0
#define	MXNET_USE_INT64_TENSOR_SIZE   MSHADOW_INT64_TENSOR_SIZE

Enumerations
enum	: unsigned {   mxnet::features::CUDA = 0, mxnet::features::CUDNN, mxnet::features::NCCL, mxnet::features::CUDA_RTC,   mxnet::features::TENSORRT, mxnet::features::CPU_SSE, mxnet::features::CPU_SSE2, mxnet::features::CPU_SSE3,   mxnet::features::CPU_SSE4_1, mxnet::features::CPU_SSE4_2, mxnet::features::CPU_SSE4A, mxnet::features::CPU_AVX,   mxnet::features::CPU_AVX2, mxnet::features::OPENMP, mxnet::features::SSE, mxnet::features::F16C,   mxnet::features::JEMALLOC, mxnet::features::BLAS_OPEN, mxnet::features::BLAS_ATLAS, mxnet::features::BLAS_MKL,   mxnet::features::BLAS_APPLE, mxnet::features::LAPACK, mxnet::features::MKLDNN, mxnet::features::OPENCV,   mxnet::features::CAFFE, mxnet::features::PROFILER, mxnet::features::DIST_KVSTORE, mxnet::features::CXX14,   mxnet::features::INT64_TENSOR_SIZE, mxnet::features::SIGNAL_HANDLER, mxnet::features::DEBUG, mxnet::features::MAX_FEATURES }
	Compile time features. More...

Functions
bool	mxnet::features::is_enabled (unsigned feat)

Detailed Description

get features of the MXNet library at runtime

Copyright (c) 2018 by Contributors

Author

larroy

Macro Definition Documentation

#define MXNET_ENABLE_CUDA_RTC   0

#define MXNET_GPU_NOT_ENABLED_ERROR   "GPU is not enabled"

Error message for using gpu when MXNET_USE_CUDA==0.

#define MXNET_USE_BLAS_APPLE   0

#define MXNET_USE_BLAS_ATLAS   0

#define MXNET_USE_BLAS_MKL   0

#define MXNET_USE_BLAS_OPEN   0

#define MXNET_USE_CAFFE   0

#define MXNET_USE_CUDNN   MSHADOW_USE_CUDNN

whether to use cuda support

whether to use cudnn library for convolution

#define MXNET_USE_CUSOLVER   MSHADOW_USE_CUSOLVER

whether to use cusolver library

#define MXNET_USE_DIST_KVSTORE   0

#define MXNET_USE_F16C   MSHADOW_USE_F16C

#define MXNET_USE_INT64_TENSOR_SIZE   MSHADOW_INT64_TENSOR_SIZE

#define MXNET_USE_LAPACK   0

#define MXNET_USE_MKLDNN   0

#define MXNET_USE_NCCL   0

#define MXNET_USE_OPENCV   0

whether to use opencv support

#define MXNET_USE_OPENMP   0

#define MXNET_USE_SIGNAL_HANDLER   0

#define MXNET_USE_TENSORRT   0