Skip to content

GitLab

"Deepspeed/Megatron-LM/mpu/vscode:/vscode.git/clone" did not exist on "316d3f9088d2479a79ff4251090254be628ebdc8"
Commit fd66c409 authored by littletomatodonkey's avatar littletomatodonkey
Browse files

rm mkl

parent 51a8d233
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
deploy/cpp_infer/src/ocr_det.cpp
View file @ fd66c409
......@@ -25,7 +25,7 @@ void DBDetector::LoadModel(const std::string &model_dir) {
config.EnableUseGpu(this->gpu_mem_, this->gpu_id_);
} else {
config.DisableGpu();
config.EnableMKLDNN(); // 开启MKLDNN加速
// config.EnableMKLDNN(); // not sugesteed to use for now
config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_);
}
......
deploy/cpp_infer/src/ocr_rec.cpp
View file @ fd66c409
......@@ -128,7 +128,7 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) {
config.EnableUseGpu(this->gpu_mem_, this->gpu_id_);
} else {
config.DisableGpu();
config.EnableMKLDNN(); // 开启MKLDNN加速
// config.EnableMKLDNN(); // not sugesteed to use for now
config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_);
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment