Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 b54fcfae 编写于 作者: J Jacek Czaja
浏览文件
操作

- fix to UT

上级 b2642609
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc
浏览文件 @ b54fcfae
...@@ -34,7 +34,7 @@ USE_OP_DEVICE_KERNEL(relu, MKLDNN); ...@@ -34,7 +34,7 @@ USE_OP_DEVICE_KERNEL(relu, MKLDNN);
USE_OP(softmax); USE_OP(softmax);
USE_OP_DEVICE_KERNEL(softmax, MKLDNN); USE_OP_DEVICE_KERNEL(softmax, MKLDNN);
USE_OP(conv2d); USE_OP(conv2d);
USE_OP_DEVICE_KERNEL_WITH_CUSTOM_TYPE(conv2d, MKLDNN, FP32); USE_OP_DEVICE_KERNEL_WITH_CUSTOM_TYPE(conv2d, MKLDNN, FP32);
namespace paddle { namespace paddle {
namespace operators { namespace operators {
...@@ -149,7 +149,7 @@ TEST(test_conv2d_noreuse_cache, cpu_place) { ...@@ -149,7 +149,7 @@ TEST(test_conv2d_noreuse_cache, cpu_place) {
CacheTester ct; CacheTester ct;
RunOperator<float>(p, "conv2d", dims, "input_signal"); RunOperator<float>(p, "conv2d", dims, "input_signal");
RunOperator<float>(p, "conv2d", dims, "input_signal2"); RunOperator<float>(p, "conv2d", dims, "input_signal2");
PADDLE_ENFORCE_EQ(ct.Analyze(9), true, PADDLE_ENFORCE_EQ(ct.Analyze(18), true,
platform::errors::InvalidArgument( platform::errors::InvalidArgument(
"Wrong number of cached oneDNN objects")); "Wrong number of cached oneDNN objects"));
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或