OSSInnovation
/
mindspore
0
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

optimize opencl conv2d on mate40

This commit is contained in:
wangdongxu 2021-03-30 18:56:24 +08:00
parent 2ad6a2d9cf
commit 306761da3b
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;