- fix

40b9025d · Jacek Czaja · e8882323 · 40b9025d
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc paddle/fluid/operators/mkldnn/test_mkldnn_caching.cc +1 -1

未找到文件。
--- 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 {