mxnet
Classes | Namespaces | Macros | Enumerations | Functions
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