diff options
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.cpp | 5 |
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; |