Skip s8u8 when force fp32

d8181ab9 · Haihao Shen · 1d9d8ed6 · d8181ab9
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/conv_mkldnn_op.cc paddle/fluid/operators/conv_mkldnn_op.cc +1 -1

未找到文件。
--- a/paddle/fluid/operators/conv_mkldnn_op.cc
+++ b/paddle/fluid/operators/conv_mkldnn_op.cc
@@ -785,7 +785,7 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
    pipeline.push_back(*conv_p);
    stream(stream::kind::eager).submit(pipeline).wait();
-    if(need_s8_to_u8){
+    if(need_s8_to_u8 && !force_fp32_output){
        output->mutable_data<uint8_t>(ctx.GetPlace());
    }