From 40b9025de58a1ee9e7b55ee50d31d39cee1a0f45 Mon Sep 17 00:00:00 2001 From: Jacek Czaja Date: Thu, 5 Aug 2021 17:20:56 +0200 Subject: [PATCH] - fix --- paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc b/paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc index 60e1b45fd4d..56971b9443f 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(conv2d, MKLDNN); +USE_OP_DEVICE_KERNEL_WITH_CUSTOM_TYPE(conv2d, MKLDNN, FP32); namespace paddle { namespace operators { -- GitLab