!13000 Register ModelImpl with compile macro for version check

From: @wilfchen
Reviewed-by: @limingqi107,@cristoval
Signed-off-by: @cristoval
pull/13000/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 34c0418ac1

@ -21,8 +21,12 @@
#include "cxx_api/factory.h"
namespace mindspore {
// mindspore-serving check current package for version check with ModelImpl factory.
#if ENABLE_D
API_FACTORY_REG(ModelImpl, Ascend910, MsModel);
#elif ENABLE_GPU
API_FACTORY_REG(ModelImpl, GPU, MsModel);
#endif
static std::string GenerateShapeKey(const std::vector<std::vector<int64_t>> &dims) {
std::string shape_key;

Loading…
Cancel
Save