From b6255ab9e797c62be0b4cdde836d4436e21decd4 Mon Sep 17 00:00:00 2001
From: Alexander Alekhin <alexander.alekhin@intel.com>
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 b74bf4d8e8..8b74248b64 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<Dtype>::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.18.0