Browse Source

!10027 Dvpp debug

From: @lizhenglong1992
Reviewed-by: @zhoufeng54,@pandoublefeng
Signed-off-by: @pandoublefeng
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
18eb589daa
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/dvpp_decode_resize_crop_jpeg_op.cc

+ 2
- 1
mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/dvpp_decode_resize_crop_jpeg_op.cc View File

@@ -72,7 +72,7 @@ Status DvppDecodeResizeCropJpegOp::Compute(const std::shared_ptr<Tensor> &input,
// Third part end where we execute the core function of dvpp // Third part end where we execute the core function of dvpp
auto data = std::static_pointer_cast<unsigned char>(process.Get_Memory_Data()); auto data = std::static_pointer_cast<unsigned char>(process.Get_Memory_Data());
unsigned char *ret_ptr = data.get(); unsigned char *ret_ptr = data.get();
std::shared_ptr(DvppDataInfo) CropOut = process.Get_Device_Memory_Data();
std::shared_ptr<DvppDataInfo> CropOut = process.Get_Device_Memory_Data();
dsize_t dvpp_length = CropOut->dataSize; dsize_t dvpp_length = CropOut->dataSize;
const TensorShape dvpp_shape({dvpp_length, 1, 1}); const TensorShape dvpp_shape({dvpp_length, 1, 1});
const DataType dvpp_data_type(DataType::DE_UINT8); const DataType dvpp_data_type(DataType::DE_UINT8);
@@ -82,6 +82,7 @@ Status DvppDecodeResizeCropJpegOp::Compute(const std::shared_ptr<Tensor> &input,
RETURN_STATUS_UNEXPECTED(error); RETURN_STATUS_UNEXPECTED(error);
} }
process.device_memory_release(); 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. // Last part end where we transform the processed data into a tensor which can be applied in later units.
} catch (const cv::Exception &e) { } catch (const cv::Exception &e) {
std::string error = "[ERROR] Fail in DvppDecodeResizeCropJpegOp:" + std::string(e.what()); std::string error = "[ERROR] Fail in DvppDecodeResizeCropJpegOp:" + std::string(e.what());


Loading…
Cancel
Save