From 4655f5bb75a0553ba5f203a74293409cd1816c4a Mon Sep 17 00:00:00 2001 From: Zhenglong Li Date: Wed, 16 Dec 2020 09:53:44 +0800 Subject: [PATCH] Fix one complie problem and make dvpp feature robust. --- .../kernels/image/dvpp/dvpp_decode_resize_crop_jpeg_op.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/dvpp_decode_resize_crop_jpeg_op.cc b/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/dvpp_decode_resize_crop_jpeg_op.cc index a74f3d6479..35e7c7cdb5 100644 --- a/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/dvpp_decode_resize_crop_jpeg_op.cc +++ b/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/dvpp_decode_resize_crop_jpeg_op.cc @@ -72,7 +72,7 @@ Status DvppDecodeResizeCropJpegOp::Compute(const std::shared_ptr &input, // Third part end where we execute the core function of dvpp auto data = std::static_pointer_cast(process.Get_Memory_Data()); unsigned char *ret_ptr = data.get(); - std::shared_ptr(DvppDataInfo) CropOut = process.Get_Device_Memory_Data(); + std::shared_ptr CropOut = process.Get_Device_Memory_Data(); dsize_t dvpp_length = CropOut->dataSize; const TensorShape dvpp_shape({dvpp_length, 1, 1}); const DataType dvpp_data_type(DataType::DE_UINT8); @@ -82,6 +82,7 @@ Status DvppDecodeResizeCropJpegOp::Compute(const std::shared_ptr &input, RETURN_STATUS_UNEXPECTED(error); } process.device_memory_release(); + process.Release(); // Last part end where we transform the processed data into a tensor which can be applied in later units. } catch (const cv::Exception &e) { std::string error = "[ERROR] Fail in DvppDecodeResizeCropJpegOp:" + std::string(e.what());