| 35 | } |
| 36 | |
| 37 | void LoadCustomEngineLib(const std::string& dso_lib_path, |
| 38 | CustomEngineParams* engine_params) { |
| 39 | if (ValidCustomCustomEngineParams(engine_params)) { |
| 40 | paddle::custom_engine::CustomEngineManager::Instance() |
| 41 | ->SetCustomEngineInterface(engine_params->interface); |
| 42 | |
| 43 | auto* interface = paddle::custom_engine::CustomEngineManager::Instance() |
| 44 | ->GetCustomEngineInterface(); |
| 45 | |
| 46 | // register custom engine op |
| 47 | if (interface->register_custom_engine_op) { |
| 48 | interface->register_custom_engine_op(); |
| 49 | } else { |
| 50 | LOG(WARNING) << "Skipped lib [" << dso_lib_path |
| 51 | << "]. register_custom_engine_op is not set!!! please check " |
| 52 | "the version " |
| 53 | "compatibility between PaddlePaddle and Custom Engine."; |
| 54 | } |
| 55 | |
| 56 | } else { |
| 57 | LOG(WARNING) << "Skipped lib [" << dso_lib_path |
| 58 | << "]. Wrong engine parameters!!! please check the version " |
| 59 | "compatibility between PaddlePaddle and Custom Engine."; |
| 60 | } |
| 61 | } |
| 62 | |
| 63 | } // namespace custom_engine |
| 64 | } // namespace paddle |