Browse Source

feat(lite): add set opencl buffer kernel cache lite api

GitOrigin-RevId: 0605218238
release-1.10
Megvii Engine Team 3 years ago
parent
commit
ba32360af4
2 changed files with 3 additions and 2 deletions
  1. +1
    -0
      lite/load_and_run/src/models/model_lite.cpp
  2. +2
    -2
      lite/load_and_run/src/models/model_lite.h

+ 1
- 0
lite/load_and_run/src/models/model_lite.cpp View File

@@ -13,6 +13,7 @@ ModelLite::ModelLite(const std::string& path) : model_path(path) {
void ModelLite::load_model() { void ModelLite::load_model() {
m_network = std::make_shared<lite::Network>(config, IO); m_network = std::make_shared<lite::Network>(config, IO);
if (enable_layout_transform) { if (enable_layout_transform) {
LITE_WARN("enable layout transform while load model for lite");
lite::Runtime::enable_global_layout_transform(m_network); lite::Runtime::enable_global_layout_transform(m_network);
} }
if (share_model_mem) { if (share_model_mem) {


+ 2
- 2
lite/load_and_run/src/models/model_lite.h View File

@@ -60,8 +60,8 @@ public:
std::vector<uint8_t> get_model_data() override; std::vector<uint8_t> get_model_data() override;


private: private:
bool share_model_mem;
bool enable_layout_transform;
bool share_model_mem = false;
bool enable_layout_transform = false;
std::string model_path; std::string model_path;


DataParser parser; DataParser parser;


Loading…
Cancel
Save