Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!14399 【MS】【LITE】【GPU】 optimize opencl performance on mate40 

From: @wangdongxu6
Reviewed-by: @ddwsky,@hangangqiang
Signed-off-by: @ddwsky
This commit is contained in:
mindspore-ci-bot 2021-03-31 09:22:43 +08:00 committed by Gitee
parent 1edbbe56ba 8c3e9e49d2
commit 47812ff252
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/lite/src/runtime/kernel/opencl/kernel/conv2d.cc
View File

@ -350,6 +350,9 @@ void Conv2DOpenCLKernel::SetGlobalLocal() {
size_t global_w = UP_DIV(OW_, block_size_.W);
size_t global_c = UP_DIV(CO_SLICES_, block_size_.C);
int local_max = filter_type_ == MemType::IMG ? 64 : 128;
if (ocl_runtime_->DeviceComputeUnits() > 16) {
local_max = 256;
}
const int local_c_max = 16;
const int OH_threshold = 100;
const int OW_threshold = 100;