Skip to content

Commit 08c96d1

Browse files
remove mkldnn & fix commit
test=develop
1 parent 725b98f commit 08c96d1

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

paddle/fluid/operators/data_norm_op.cc

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,9 @@ limitations under the License. */
1515
#include "paddle/fluid/operators/data_norm_op.h"
1616
#include <string>
1717
#include "paddle/fluid/framework/data_layout.h"
18+
#ifdef PADDLE_WITH_MKLDNN
19+
#include "paddle/fluid/platform/mkldnn_helper.h"
20+
#endif
1821

1922
namespace paddle {
2023
namespace operators {
@@ -94,6 +97,13 @@ class DataNormOp : public framework::OperatorWithKernel {
9497
// TODO(pzelazko-intel): enable MKLDNN layout when it's ready
9598
framework::LibraryType library = framework::LibraryType::kPlain;
9699
framework::DataLayout layout = framework::DataLayout::kAnyLayout;
100+
#ifdef PADDLE_WITH_MKLDNN
101+
if (library == framework::LibraryType::kPlain &&
102+
platform::CanMKLDNNBeUsed(ctx)) {
103+
library = framework::LibraryType::kMKLDNN;
104+
layout = framework::DataLayout::kMKLDNN;
105+
}
106+
#endif
97107

98108
return framework::OpKernelType(input_data_type, ctx.GetPlace(), layout,
99109
library);
@@ -251,6 +261,14 @@ class DataNormGradOp : public framework::OperatorWithKernel {
251261
framework::LibraryType library = framework::LibraryType::kPlain;
252262
framework::DataLayout layout = framework::DataLayout::kAnyLayout;
253263

264+
#ifdef PADDLE_WITH_MKLDNN
265+
if (library == framework::LibraryType::kPlain &&
266+
platform::CanMKLDNNBeUsed(ctx)) {
267+
library = framework::LibraryType::kMKLDNN;
268+
layout = framework::DataLayout::kMKLDNN;
269+
}
270+
#endif
271+
254272
return framework::OpKernelType(ctx.Input<Tensor>("X")->type(),
255273
ctx.GetPlace(), layout, library);
256274
}

0 commit comments

Comments
 (0)