From c7b80a1fe6ccd5a4946b239335e8d418ea3864ce Mon Sep 17 00:00:00 2001 From: chuxing Date: Sat, 26 Dec 2020 13:52:10 +0800 Subject: [PATCH] fixing SC --- ge/hybrid/model/hybrid_model.cc | 2 +- ge/hybrid/model/hybrid_model_builder.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ge/hybrid/model/hybrid_model.cc b/ge/hybrid/model/hybrid_model.cc index c24f2f5e..91b6a549 100644 --- a/ge/hybrid/model/hybrid_model.cc +++ b/ge/hybrid/model/hybrid_model.cc @@ -341,7 +341,7 @@ TensorValue *HybridModel::GetConstant(const NodePtr &node) const { return it->second.get(); } -TensorValue * HybridModel::GetTensor(const NodePtr &node) const { +TensorValue *HybridModel::GetTensor(const NodePtr &node) const { if (node == nullptr) { GELOGE(PARAM_INVALID, "Param is null"); return nullptr; diff --git a/ge/hybrid/model/hybrid_model_builder.cc b/ge/hybrid/model/hybrid_model_builder.cc index d89653c0..58f4723d 100755 --- a/ge/hybrid/model/hybrid_model_builder.cc +++ b/ge/hybrid/model/hybrid_model_builder.cc @@ -944,9 +944,9 @@ Status HybridModelBuilder::InitWeights() { constant_node->GetName().c_str(), tensor_size); - std::shared_ptr tensor_buffer(TensorBuffer::Create(allocator, tensor_size).release()); + auto tensor_buffer = TensorBuffer::Create(allocator, tensor_size); GE_CHECK_NOTNULL(tensor_buffer); - std::unique_ptr constant_tensor(new (std::nothrow)TensorValue(tensor_buffer)); + std::unique_ptr constant_tensor(new (std::nothrow)TensorValue(std::move(tensor_buffer))); GE_CHECK_NOTNULL(constant_tensor); constant_tensor->SetName("Constant_" + op_desc->GetName()); if (tensor_size > 0) {