summaryrefslogtreecommitdiff
path: root/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/conv/deconv.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'inference-engine/thirdparty/mkl-dnn/tests/benchdnn/conv/deconv.cpp')
-rw-r--r--inference-engine/thirdparty/mkl-dnn/tests/benchdnn/conv/deconv.cpp5
1 files changed, 4 insertions, 1 deletions
diff --git a/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/conv/deconv.cpp b/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/conv/deconv.cpp
index 7785b6f60..ec0e0d04d 100644
--- a/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/conv/deconv.cpp
+++ b/inference-engine/thirdparty/mkl-dnn/tests/benchdnn/conv/deconv.cpp
@@ -41,7 +41,7 @@ inline static void swap(int &a, int &b)
}
inline bool is_deconv_3d(const prb_t *p)
{
- return (p->id > 1) ? 1 : 0;
+ return (p->id > 1 || p->od > 1) ? 1 : 0;
}
inline int transpose_data_wei(const prb_t *p, dnn_mem_t &wei, dnn_mem_t &wei_tr) {
@@ -299,6 +299,9 @@ int doit(const prb_t *p, res_t *r) {
}
}
+ DNN_SAFE_V(mkldnn_primitive_destroy(c));
+ DNN_SAFE_V(mkldnn_primitive_desc_destroy(dpd));
+
delete p_bia_dt;
delete p_bia_fp;
delete p_zero_fp;