mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix gpu bug

This commit is contained in:
yeyunpeng2020 2021-07-13 15:58:48 +08:00
parent a51991e777
commit b580a2122f
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/lite/src/common/context_util.cc
View File

@ -42,13 +42,12 @@ std::shared_ptr<mindspore::CPUDeviceInfo> CPUDeviceInfoFromCPUDeviceContext(cons
return cpu_info; return cpu_info;
} }
std::shared_ptr<mindspore::MaliGPUDeviceInfo> GPUDeviceInfoFromGPUDeviceContext( std::shared_ptr<mindspore::GPUDeviceInfo> GPUDeviceInfoFromGPUDeviceContext(const lite::DeviceContext &gpu_context) {
const lite::DeviceContext &gpu_context) {
if (gpu_context.device_type_ != DT_GPU) { if (gpu_context.device_type_ != DT_GPU) {
MS_LOG(ERROR) << "function input parameter is not gpu context."; MS_LOG(ERROR) << "function input parameter is not gpu context.";
return nullptr; return nullptr;
} }
auto gpu_info = std::make_shared<mindspore::MaliGPUDeviceInfo>(); auto gpu_info = std::make_shared<mindspore::GPUDeviceInfo>();
gpu_info->SetEnableFP16(gpu_context.device_info_.gpu_device_info_.enable_float16_); gpu_info->SetEnableFP16(gpu_context.device_info_.gpu_device_info_.enable_float16_);
PassBasicProperties(gpu_info, gpu_context); PassBasicProperties(gpu_info, gpu_context);
return gpu_info; return gpu_info;