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

!23640 fix opencl allocator bug 

Merge pull request !23640 from yeyunpeng2020/gpu
This commit is contained in:
i-robot 2021-09-17 02:40:31 +00:00 committed by Gitee
parent 80e486c219 fe024115fd
commit a7f17d5c8e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/lite/src/runtime/gpu/opencl/opencl_executor.cc
View File

@ -108,6 +108,11 @@ int OpenCLExecutor::Tune(kernel::OpenCLKernel *op_kernel) {
MS_LOG(ERROR) << "tuning kernel failed, name: " << op_kernel->name();
return ret;
}
ret = op_kernel->PostProcess();
if (ret != RET_OK) {
MS_LOG(WARNING) << "PostProcess kernel failed, name: " << op_kernel->name() << " in tuning";
return ret;
}
return RET_OK;
}
} // namespace mindspore::lite::opencl