mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-07 09:25:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #11941 from alalek:dnn_ocl_fix_verify_umat_mapping

Browse Source
This commit is contained in:
Alexander Alekhin 2018-07-12 07:37:55 +00:00
commit 9c3ec05825
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/dnn/src/ocl4dnn/src/ocl4dnn_conv_spatial.cpp
View File

@ -1230,20 +1230,22 @@ bool OCL4DNNConvSpatial<float>::verifyResult(const UMat &bottom,
tuned_ = saved_tuned; tuned_ = saved_tuned;
UMat new_top, new_verify_top; UMat new_top, new_verify_top;
float *data, *verify_data; Mat mat_top, mat_verify_top;
if (use_half_) if (use_half_)
{ {
convertFp16(top, new_top); convertFp16(top, new_top);
convertFp16(verifyTop, new_verify_top); convertFp16(verifyTop, new_verify_top);
data = (float *)new_top.getMat(ACCESS_READ).ptr<float>(); mat_top = new_top.getMat(ACCESS_READ);
verify_data = (float *)new_verify_top.getMat(ACCESS_READ).ptr<float>(); mat_verify_top = new_verify_top.getMat(ACCESS_READ);
} }
else else
{ {
data = (float *)top.getMat(ACCESS_READ).ptr<float>(); mat_top = top.getMat(ACCESS_READ);
verify_data = (float *)verifyTop.getMat(ACCESS_READ).ptr<float>(); mat_verify_top = verifyTop.getMat(ACCESS_READ);
} }
const float* data = mat_top.ptr<float>();
const float* verify_data = mat_verify_top.ptr<float>();
for (int32_t n = 0; n < num_; ++n) { for (int32_t n = 0; n < num_; ++n) {
for (int32_t g = 0; g < group_; ++g) { for (int32_t g = 0; g < group_; ++g) {