From e4fa3c046cf01c6c7fa255cedf2f4ba2bf6f0e27 Mon Sep 17 00:00:00 2001 From: liuqi Date: Mon, 9 Apr 2018 14:50:44 +0800 Subject: [PATCH] Fix bug: load opencl so when run on cpu. --- mace/core/mace.cc | 4 +++- mace/tools/validation/mace_run.cc | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/mace/core/mace.cc b/mace/core/mace.cc index 484c9472..5f7647ed 100644 --- a/mace/core/mace.cc +++ b/mace/core/mace.cc @@ -176,7 +176,9 @@ MaceStatus MaceEngine::Impl::Run( LOG(FATAL) << "Net run failed"; } } - OpenCLRuntime::Global()->SaveBuiltCLProgram(); + if (device_type_ == OPENCL) { + OpenCLRuntime::Global()->SaveBuiltCLProgram(); + } for (auto &output : *outputs) { Tensor *output_tensor = ws_->GetTensor(MakeString("mace_output_node_", output.first + ":0")); diff --git a/mace/tools/validation/mace_run.cc b/mace/tools/validation/mace_run.cc index d4fa7c29..60f5cd59 100644 --- a/mace/tools/validation/mace_run.cc +++ b/mace/tools/validation/mace_run.cc @@ -308,7 +308,9 @@ bool RunModel(const std::vector &input_names, printf("time %11.3f %11.3f %11.3f %11.3f %11.3f\n", create_net_millis, mace_engine_ctor_millis, init_millis, warmup_millis, model_run_millis); - WriteOpenCLPlatformInfo(kernel_file_path); + if (device_type == DeviceType::OPENCL) { + WriteOpenCLPlatformInfo(kernel_file_path); + } for (size_t i = 0; i < output_count; ++i) { std::string output_name = -- GitLab