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

!17737 Remove inference with trt switch 

From: @wilfchen
Reviewed-by: @cristoval,@limingqi107
Signed-off-by: @limingqi107
This commit is contained in:
mindspore-ci-bot 2021-06-05 09:25:33 +08:00 committed by Gitee
parent e6c16a185d 1acd3b47fa
commit 3914d09ce8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/cxx_api/graph/gpu/gpu_graph_impl.cc
View File

@ -72,7 +72,7 @@ Status GPUGraphImpl::InitEnv() {
if (gpu_info == nullptr) {
return kMCDeviceError;
}
ms_context->set_param<bool>(MS_CTX_ENABLE_INFER_OPT, gpu_info->GetGpuTrtInferMode());
ms_context->set_param<bool>(MS_CTX_ENABLE_INFER_OPT, true);
ms_context->set_param<std::string>(MS_CTX_INFER_PRECISION_MODE, gpu_info->GetPrecisionMode());
session_impl_ = session::SessionFactory::Get().Create(kGpuInferenceDevice);