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
Compile time feature description, member fields: name and enabled. |
|
OrderedDict of name to Feature |
Functions
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
True if MXNet was compiled with the given compile-time feature.
Feature name.
-
property
enabled
¶ True if MXNet was compiled with the given compile-time feature.
-
property
name
¶ Feature name.
-
property
-
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
Did this page help you?
Yes
No
Thanks for your feedback!