MXNet

API

 mxnet / mxnet.runtime


mxnet.runtime

Runtime querying of compile time features in the native library.

With this module you can check at runtime which libraries and features were compiled in the library.

Example usage:

import mxnet
features=mxnet.runtime.Features()

features.is_enabled("CUDNN")
False

features.is_enabled("CPU_SSE")
True

print(features)
[✖ CUDA, ✖ CUDNN, ✖ NCCL, ✖ CUDA_RTC, ✖ TENSORRT, ✔ CPU_SSE, ✔ CPU_SSE2, ✔ CPU_SSE3,
✔ CPU_SSE4_1, ✔ CPU_SSE4_2, ✖ CPU_SSE4A, ✔ CPU_AVX, ✖ CPU_AVX2, ✔ OPENMP, ✖ SSE,
✔ F16C, ✔ JEMALLOC, ✔ BLAS_OPEN, ✖ BLAS_ATLAS, ✖ BLAS_MKL, ✖ BLAS_APPLE, ✔ LAPACK,
✖ MKLDNN, ✔ OPENCV, ✖ CAFFE, ✖ PROFILER, ✖ DIST_KVSTORE, ✖ CXX14, ✖ INT64_TENSOR_SIZE,
✔ SIGNAL_HANDLER, ✔ DEBUG, ✖ TVM_OP]

Classes

Feature

Compile time feature description, member fields: name and enabled.

Features

OrderedDict of name to Feature

Functions

feature_list()

Check the library for compile-time features.

class mxnet.runtime.Feature[source]

Bases: _ctypes.Structure

Compile time feature description, member fields: name and enabled.

Attributes

enabled

True if MXNet was compiled with the given compile-time feature.

name

Feature name.

property enabled

True if MXNet was compiled with the given compile-time feature.

property name

Feature name.

class mxnet.runtime.Features[source]

Bases: collections.OrderedDict

OrderedDict of name to Feature

Methods

is_enabled(feature_name)

Check for a particular feature by name

is_enabled(feature_name)[source]

Check for a particular feature by name

Parameters

feature_name (str) – The name of a valid feature as string for example ‘CUDA’

Returns

True if it’s enabled, False if it’s disabled, RuntimeError if the feature is not known

Return type

Boolean

mxnet.runtime.feature_list()[source]

Check the library for compile-time features. The list of features are maintained in libinfo.h and libinfo.cc

Returns

List of Feature objects

Return type

list


此页内容是否对您有帮助