Skip to content

Commit 8d5ab1f

Browse files
authored
Merge pull request #11595 from luotao1/refine_code
add url of cuda9.0_cudnn7_avx_mkl library
2 parents 1d4d8de + 5aac910 commit 8d5ab1f

File tree

4 files changed

+17
-30
lines changed

4 files changed

+17
-30
lines changed

doc/fluid/howto/inference/build_and_install_lib_cn.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ cpu_noavx_openblas `fluid.tgz <https://guest:@paddleci.ngrok.io/repository
1313
cuda7.5_cudnn5_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda75cudnn5cp27cp27mu/.lastSuccessful/fluid.tgz>`_
1414
cuda8.0_cudnn5_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda80cudnn5cp27cp27mu/.lastSuccessful/fluid.tgz>`_
1515
cuda8.0_cudnn7_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda8cudnn7cp27cp27mu/.lastSuccessful/fluid.tgz>`_
16+
cuda9.0_cudnn7_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda90cudnn7avxMkl/.lastSuccessful/fluid.tgz>`_
1617
====================== ========================================
1718

1819
从源码编译

paddle/fluid/operators/batch_norm_mkldnn_op.cc

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,6 @@ namespace operators {
2121

2222
using batch_norm_bwd = mkldnn::batch_normalization_backward;
2323
using batch_norm_fwd = mkldnn::batch_normalization_forward;
24-
using framework::DataLayout;
25-
using framework::Tensor;
2624
using mkldnn::memory;
2725
using mkldnn::primitive;
2826
using mkldnn::reorder;
@@ -31,18 +29,6 @@ using paddle::platform::MKLDNNDeviceContext;
3129
using paddle::platform::MKLDNNMemDesc;
3230
using platform::to_void_cast;
3331

34-
template <typename T>
35-
using EigenArrayMap =
36-
Eigen::Map<Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
37-
template <typename T>
38-
using ConstEigenArrayMap =
39-
Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
40-
template <typename T>
41-
using EigenVectorArrayMap = Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1>>;
42-
template <typename T>
43-
using ConstEigenVectorArrayMap =
44-
Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, 1>>;
45-
4632
namespace {
4733
template <typename T>
4834
struct bn_type_traits {

paddle/fluid/operators/batch_norm_op.cc

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -22,22 +22,6 @@ limitations under the License. */
2222
namespace paddle {
2323
namespace operators {
2424

25-
using Tensor = framework::Tensor;
26-
using LoDTensor = framework::LoDTensor;
27-
using DataLayout = framework::DataLayout;
28-
29-
template <typename T>
30-
using EigenArrayMap =
31-
Eigen::Map<Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
32-
template <typename T>
33-
using ConstEigenArrayMap =
34-
Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
35-
template <typename T>
36-
using EigenVectorArrayMap = Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1>>;
37-
template <typename T>
38-
using ConstEigenVectorArrayMap =
39-
Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, 1>>;
40-
4125
class BatchNormOp : public framework::OperatorWithKernel {
4226
public:
4327
using framework::OperatorWithKernel::OperatorWithKernel;

paddle/fluid/operators/batch_norm_op.h

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,22 @@ limitations under the License. */
1919
namespace paddle {
2020
namespace operators {
2121

22+
using Tensor = framework::Tensor;
23+
using LoDTensor = framework::LoDTensor;
24+
using DataLayout = framework::DataLayout;
25+
26+
template <typename T>
27+
using EigenArrayMap =
28+
Eigen::Map<Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
29+
template <typename T>
30+
using ConstEigenArrayMap =
31+
Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
32+
template <typename T>
33+
using EigenVectorArrayMap = Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1>>;
34+
template <typename T>
35+
using ConstEigenVectorArrayMap =
36+
Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, 1>>;
37+
2238
template <typename DeviceContext, typename T>
2339
class BatchNormKernel : public framework::OpKernel<T> {
2440
public:

0 commit comments

Comments
 (0)