Caffe2 - C++ API
A deep learning, cross platform ML framework
mkl_version_check.h
1 
17 #ifndef CAFFE2_UTILS_MKL_MKL_VERSION_CHECK_H_
18 #define CAFFE2_UTILS_MKL_MKL_VERSION_CHECK_H_
19 #ifdef CAFFE2_USE_MKL
20 
21 #include <mkl.h>
22 
23 #if INTEL_MKL_VERSION >= 20170000
24 #define CAFFE2_HAS_MKL_SGEMM_PACK
25 #define CAFFE2_HAS_MKL_DNN
26 #endif // INTEL_MKL_VERSION >= 20170000
27 
28 #endif // CAFFE2_USE_MKL
29 #endif // CAFFE2_UTILS_MKL_MKL_VERSION_CHECK_H_