diff --git a/core/predictor/framework/infer.cpp b/core/predictor/framework/infer.cpp old mode 100644 new mode 100755 index 871e666d5c37ae46044aee8197685c9518391193..fa77c50506c02c4350569349e4e6679dde173ffc --- a/core/predictor/framework/infer.cpp +++ b/core/predictor/framework/infer.cpp @@ -391,7 +391,7 @@ int InferManager::proc_initialize(const char* path, return -1; } uint32_t engine_num = model_toolkit_conf.engines_size(); - im::bsf::TaskExecutorVector::instance().resize(engine_num); + im::bsf::TaskExecutorVector::instance().resize(*engine_index_ptr+engine_num); for (uint32_t ei = 0; ei < engine_num; ++ei) { LOG(INFO) << "model_toolkit_conf.engines(" << ei << ").name: " << model_toolkit_conf.engines(ei).name(); diff --git a/core/predictor/framework/infer.h b/core/predictor/framework/infer.h index 469737e3b21e9b5cd3f9e0eac55f351da9c81eca..93be13c684874b8b5a6686f3aeddd2942037d84c 100755 --- a/core/predictor/framework/infer.h +++ b/core/predictor/framework/infer.h @@ -18,11 +18,11 @@ #include #include #include +#include #include #include #include #include -#include #include "core/predictor/common/inner_common.h" #include "core/predictor/framework/bsf.h" #include "core/predictor/framework/factory.h" diff --git a/core/predictor/framework/resource.cpp b/core/predictor/framework/resource.cpp old mode 100644 new mode 100755 diff --git a/core/predictor/framework/resource.h b/core/predictor/framework/resource.h index 083d1bca9dc9a2fa837f948d58dd552a310086ed..d8a114dab581b71182c1a510db16aa0d2e818b0a 100755 --- a/core/predictor/framework/resource.h +++ b/core/predictor/framework/resource.h @@ -16,8 +16,8 @@ #include #include #include -#include #include +#include #include "core/cube/cube-api/include/cube_api.h" #include "core/predictor/common/inner_common.h" #include "core/predictor/framework/infer.h"