diff --git a/paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc b/paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc index 22cf5c457b098975a75e9d85ae0706efd0cc68c1..e2fb329407dd12a69d169db2863b39f7ff22095e 100644 --- a/paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc +++ b/paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc @@ -34,7 +34,7 @@ USE_OP_DEVICE_KERNEL(relu, MKLDNN); USE_OP(softmax); USE_OP_DEVICE_KERNEL(softmax, MKLDNN); 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 operators { @@ -149,7 +149,7 @@ TEST(test_conv2d_noreuse_cache, cpu_place) { CacheTester ct; RunOperator(p, "conv2d", dims, "input_signal"); RunOperator(p, "conv2d", dims, "input_signal2"); - PADDLE_ENFORCE_EQ(ct.Analyze(9), true, + PADDLE_ENFORCE_EQ(ct.Analyze(18), true, platform::errors::InvalidArgument( "Wrong number of cached oneDNN objects")); }