diff options
| author | nihui <shuizhuyuanluo@126.com> | 2019-03-25 11:55:52 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-03-25 11:59:10 -0700 |
| commit | ed8c462dc79461c434cdd69f378743d1258c624b (patch) | |
| tree | 7c210bbe07c6edbc4f95fe6640948e4c82d7f09a /caffe2 | |
| parent | 6c9b312fd48c5a3e00719be881aa067b436d05e5 (diff) | |
| download | pytorch-ed8c462dc79461c434cdd69f378743d1258c624b.tar.gz pytorch-ed8c462dc79461c434cdd69f378743d1258c624b.tar.bz2 pytorch-ed8c462dc79461c434cdd69f378743d1258c624b.zip | |
Fix caffe2 build with BLAS=OpenBLAS (#18422)
Summary:
g++ complains about failing to find the declaration of cblas_sscal and cblas_dscal BLAS function
let's fix it :)
fedora 29, gcc 8.3.1, openblas 0.3.5
build with cmake -DBLAS=OpenBLAS ..
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18422
Differential Revision: D14598977
Pulled By: soumith
fbshipit-source-id: bde77bfb359d2ff38226401caeed78c114ef7468
Diffstat (limited to 'caffe2')
| -rw-r--r-- | caffe2/utils/math/elementwise.cc | 1 | ||||
| -rw-r--r-- | caffe2/utils/math/reduce.cc | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/caffe2/utils/math/elementwise.cc b/caffe2/utils/math/elementwise.cc index 96e4644b82..7e56c26d5a 100644 --- a/caffe2/utils/math/elementwise.cc +++ b/caffe2/utils/math/elementwise.cc @@ -13,6 +13,7 @@ #include "caffe2/core/context.h" #include "caffe2/utils/eigen_utils.h" +#include "caffe2/utils/math.h" namespace caffe2 { namespace math { diff --git a/caffe2/utils/math/reduce.cc b/caffe2/utils/math/reduce.cc index 09824df470..c71d353740 100644 --- a/caffe2/utils/math/reduce.cc +++ b/caffe2/utils/math/reduce.cc @@ -16,6 +16,7 @@ #include "caffe2/core/context.h" #include "caffe2/utils/eigen_utils.h" +#include "caffe2/utils/math.h" #include "caffe2/utils/math/elementwise.h" #include "caffe2/utils/math/utils.h" |