diff --git a/paddle/fluid/operators/conv_mkldnn_op.cc b/paddle/fluid/operators/conv_mkldnn_op.cc index 64be8a4b16314d58fb84b734088b03c61d1e79d0..6a6ef818927b3646fdb4fa6b345d5ed420376d94 100644 --- a/paddle/fluid/operators/conv_mkldnn_op.cc +++ b/paddle/fluid/operators/conv_mkldnn_op.cc @@ -350,27 +350,59 @@ std::cout<<"fuse_relu = "<1? weights_tz[1]*weights_tz[0] : weights_tz[0]) : 1; float scale_in_data = *(scale_in->data()); + +for(int i=0; i<50; i++){ + printf("%f ", (float)(input_data[i])/scale_in_data); +} +printf("\n");fflush(stdout); + +std::cout<<"scale_in_data = "< scale_weights_data(count); +std::cout<<"scale_weights_data = "; for(int i=0; idata() + i); +std::cout<data()); +std::cout<<"scale_out_data = "<data()); +std::cout<<"scale_in_eltwise_data = "<(output_data)); auto reorder_pd = std::shared_ptr( - new reorder::primitive_desc(dst_pd, src_pd, attri)); + new reorder::primitive_desc(src_pd, dst_pd, attri)); auto reorder_p= std::shared_ptr(new reorder(*reorder_pd, *src_memory_p, dst_memory)); pipeline.push_back(*reorder_p); diff --git a/paddle/fluid/operators/quantize_op.cc b/paddle/fluid/operators/quantize_op.cc index cb5a9e4c1c8d6f94e8dce9eb753fb3763625baa4..1748e645d76a79a1305dcd0c13b6dd8eb91eb0ec 100644 --- a/paddle/fluid/operators/quantize_op.cc +++ b/paddle/fluid/operators/quantize_op.cc @@ -47,9 +47,35 @@ std::cout<<"this is quantize op!!!!!!!!!!!!!!"< dst_tz = paddle::framework::vectorize2int(output->dims()); const T* input_data = input->data(); - T* output_data = output->mutable_data(ctx.GetPlace()); + uint8_t* output_data = output->mutable_data(ctx.GetPlace()); std::vector scale_data = {*(scale->data())}; +FILE *fp = fopen("quant_input.txt","w"); +printf("quantize check!!!!\n"); +std::vector tz = paddle::framework::vectorize2int(input->dims()); +int count=1; +for(int i=0; i(output_data)); + auto dst_memory = mkldnn::memory(dst_pd, to_void_cast(output_data)); auto reorder_pd = std::shared_ptr( - new reorder::primitive_desc(dst_pd, src_pd, attri)); + new reorder::primitive_desc(src_pd, dst_pd, attri)); auto reorder_p= std::shared_ptr(new reorder(*reorder_pd, *src_memory_p, dst_memory)); pipeline.push_back(*reorder_p);