From 6f0491bf09a864ada8397cec6eb030f568947b83 Mon Sep 17 00:00:00 2001
From: Huie <32028380+Huie1228@users.noreply.github.com>
Date: Tue, 3 Sep 2019 11:23:06 +0800
Subject: [PATCH] fix getgputime (#1950)

* 1.add density_prior_box for gpu.
2.add flatten2 for gpu.
3.add concat 4 inputs size for gpu.
4.fix pool.
5.fix transpose2
test=develop

* 1.fix GetGPUTime
test=develop
---
 mobile/src/io/paddle_mobile.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mobile/src/io/paddle_mobile.cpp b/mobile/src/io/paddle_mobile.cpp
index 74cac25672..95ae3763a2 100644
--- a/mobile/src/io/paddle_mobile.cpp
+++ b/mobile/src/io/paddle_mobile.cpp
@@ -451,6 +451,8 @@ double PaddleMobile<GPU_CL, float>::GetPredictTime() {
   int input_height = 224;
   int output_width = 224;
   int output_height = 224;
+  int has_group = 0;
+  int filter_channel = 3;
   status = clSetKernelArg(kernel, 0, sizeof(int), &c_block);
   CL_CHECK_ERRORS(status);
   status = clSetKernelArg(kernel, 1, sizeof(int), &w);
@@ -479,6 +481,10 @@ double PaddleMobile<GPU_CL, float>::GetPredictTime() {
   CL_CHECK_ERRORS(status);
   status = clSetKernelArg(kernel, 13, sizeof(int), &output_height);
   CL_CHECK_ERRORS(status);
+  status = clSetKernelArg(kernel, 14, sizeof(int), &filter_channel);
+  CL_CHECK_ERRORS(status);
+  status = clSetKernelArg(kernel, 15, sizeof(int), &has_group);
+  CL_CHECK_ERRORS(status);
 
   //  cl_event out_event = param.Output()->GetClEvent();
   //  cl_event wait_event = param.Input()->GetClEvent();
-- 
GitLab