From b6255ab9e797c62be0b4cdde836d4436e21decd4 Mon Sep 17 00:00:00 2001 From: Alexander Alekhin Date: Mon, 9 Jul 2018 18:02:20 +0300 Subject: [PATCH] dnn(ocl4dnn): fix args for 'max_pool_forward' kernel --- modules/dnn/src/ocl4dnn/src/ocl4dnn_pool.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modules/dnn/src/ocl4dnn/src/ocl4dnn_pool.cpp b/modules/dnn/src/ocl4dnn/src/ocl4dnn_pool.cpp index b74bf4d..8b74248 100644 --- a/modules/dnn/src/ocl4dnn/src/ocl4dnn_pool.cpp +++ b/modules/dnn/src/ocl4dnn/src/ocl4dnn_pool.cpp @@ -132,9 +132,10 @@ bool OCL4DNNPool::Forward(const UMat& bottom, width_, pooled_height_, pooled_width_, - ocl::KernelArg::PtrWriteOnly(top), - ocl::KernelArg::PtrWriteOnly(top_mask) + ocl::KernelArg::PtrWriteOnly(top) ); + if (computeMaxIdx) + oclk_max_pool_forward.set(8, ocl::KernelArg::PtrWriteOnly(top_mask)); // TODO remove magic number. Extend cv::ocl::Kernel API ret = oclk_max_pool_forward.run(1, global, local, false); } -- 2.7.4