提交 d71ed6b6 编写于 作者: 李寅

Force print mace git version

上级 c1534c96
...@@ -386,6 +386,9 @@ MaceEngine::MaceEngine(const NetDef *net_def, DeviceType device_type) ...@@ -386,6 +386,9 @@ MaceEngine::MaceEngine(const NetDef *net_def, DeviceType device_type)
net_ = std::move(CreateNet(op_registry_, *net_def, ws_.get(), device_type)); net_ = std::move(CreateNet(op_registry_, *net_def, ws_.get(), device_type));
} }
} }
extern const char *MaceGitVersion();
MaceEngine::MaceEngine(const NetDef *net_def, MaceEngine::MaceEngine(const NetDef *net_def,
DeviceType device_type, DeviceType device_type,
const std::vector<std::string> &input_nodes, const std::vector<std::string> &input_nodes,
...@@ -395,6 +398,7 @@ MaceEngine::MaceEngine(const NetDef *net_def, ...@@ -395,6 +398,7 @@ MaceEngine::MaceEngine(const NetDef *net_def,
ws_(new Workspace()), ws_(new Workspace()),
net_(nullptr), net_(nullptr),
hexagon_controller_(nullptr) { hexagon_controller_(nullptr) {
LOG(INFO) << "MACE GIT VERSION: " << MaceGitVersion();
for (auto input_name : input_nodes) { for (auto input_name : input_nodes) {
ws_->CreateTensor(MakeString("mace_input_node_", input_name, ":0"), ws_->CreateTensor(MakeString("mace_input_node_", input_name, ":0"),
GetDeviceAllocator(device_type_), DT_FLOAT); GetDeviceAllocator(device_type_), DT_FLOAT);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册