Browse Source

!15564 remove anfnodeptr in somas parameter

From: @laiyongqiang
Reviewed-by: @kisnwang,@jjfeing
Signed-off-by: @jjfeing
pull/15564/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
d340e7f102
2 changed files with 11 additions and 6 deletions
  1. +3
    -3
      mindspore/ccsrc/backend/optimizer/somas/somas.cc
  2. +8
    -3
      mindspore/ccsrc/backend/optimizer/somas/somas_parameter.h

+ 3
- 3
mindspore/ccsrc/backend/optimizer/somas/somas.cc View File

@@ -650,7 +650,8 @@ SomasParameterPtr Somas::CreateSomasParameters(AnfNodePtr node, size_t index) {
if (device_addr == nullptr) {
MS_LOG(EXCEPTION) << "Node " << node->fullname_with_scope() << " has no device address before Somas.";
}
auto param = std::make_shared<SomasParameter>(id, node, index, device_addr->GetPtr(), device_addr->GetSize());
auto param = std::make_shared<SomasParameter>(id, node->fullname_with_scope(), index, device_addr->GetPtr(),
device_addr->GetSize());
parameters_list_.push_back(param);
return param;
}
@@ -1511,8 +1512,7 @@ void Somas::DumpParameters(std::ostringstream &oss) const {
<< "\t"
<< "#" << param->size_ << "S"
<< "\t"
<< "&" << param->addr_ << "\t" << param->source_node_->fullname_with_scope() << "\t" << param->output_index_
<< "\n";
<< "&" << param->addr_ << "\t" << param->source_node_name_ << "\t" << param->output_index_ << "\n";
}
}



+ 8
- 3
mindspore/ccsrc/backend/optimizer/somas/somas_parameter.h View File

@@ -18,18 +18,23 @@
#define MINDSPORE_CCSRC_BACKEND_OPTIMIZER_SOMAS_SOMAS_PARAMETER_H_

#include <memory>
#include <string>
#include "base/base.h"

namespace mindspore {
namespace somas {
class SomasParameter {
public:
SomasParameter(size_t id, AnfNodePtr source_node, size_t index, const void *addr, size_t size)
: id_(id), source_node_(source_node), output_index_(index), addr_(const_cast<void *>(addr)), size_(size) {}
SomasParameter(size_t id, std::string source_node_name, size_t index, const void *addr, size_t size)
: id_(id),
source_node_name_(source_node_name),
output_index_(index),
addr_(const_cast<void *>(addr)),
size_(size) {}
~SomasParameter() = default;

const size_t id_{0};
AnfNodePtr source_node_;
std::string source_node_name_;
size_t output_index_;
void *addr_;
size_t size_;


Loading…
Cancel
Save