diff options
Diffstat (limited to 'include/caffe/util/mkl_alternate.hpp')
| -rw-r--r-- | include/caffe/util/mkl_alternate.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/include/caffe/util/mkl_alternate.hpp b/include/caffe/util/mkl_alternate.hpp index 3355b665..95df0f93 100644 --- a/include/caffe/util/mkl_alternate.hpp +++ b/include/caffe/util/mkl_alternate.hpp @@ -7,9 +7,14 @@ #else // If use MKL, simply include the MKL header +#ifdef USE_ACCELERATE +#include <Accelerate/Accelerate.h> +#else extern "C" { #include <cblas.h> } +#endif // USE_ACCELERATE + #include <math.h> // Functions that caffe uses but are not present if MKL is not linked. |