From ba32360af4db5b2bce5aec6217607e9e0fd3eb54 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 7 Jun 2022 13:36:13 +0800 Subject: [PATCH] feat(lite): add set opencl buffer kernel cache lite api GitOrigin-RevId: 06052182380ab11854df9ba31f50d2ad3439f692 --- lite/load_and_run/src/models/model_lite.cpp | 1 + lite/load_and_run/src/models/model_lite.h | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/lite/load_and_run/src/models/model_lite.cpp b/lite/load_and_run/src/models/model_lite.cpp index 656e1d9e..64d6833b 100644 --- a/lite/load_and_run/src/models/model_lite.cpp +++ b/lite/load_and_run/src/models/model_lite.cpp @@ -13,6 +13,7 @@ ModelLite::ModelLite(const std::string& path) : model_path(path) { void ModelLite::load_model() { m_network = std::make_shared(config, IO); if (enable_layout_transform) { + LITE_WARN("enable layout transform while load model for lite"); lite::Runtime::enable_global_layout_transform(m_network); } if (share_model_mem) { diff --git a/lite/load_and_run/src/models/model_lite.h b/lite/load_and_run/src/models/model_lite.h index ba79ed28..1df00b29 100644 --- a/lite/load_and_run/src/models/model_lite.h +++ b/lite/load_and_run/src/models/model_lite.h @@ -60,8 +60,8 @@ public: std::vector get_model_data() override; private: - bool share_model_mem; - bool enable_layout_transform; + bool share_model_mem = false; + bool enable_layout_transform = false; std::string model_path; DataParser parser;