Browse Source

upgrade ascend 0928

tags/v1.5.1
yanghaoran 3 years ago
parent
commit
7f0b65e699
3 changed files with 12 additions and 11 deletions
  1. +4
    -5
      ge/common/model/ge_model.cc
  2. +7
    -5
      ge/common/model/ge_model.h
  3. +1
    -1
      metadef

+ 4
- 5
ge/common/model/ge_model.cc View File

@@ -34,7 +34,6 @@ void GeModel::Init() {
}

GeModel::GeModel() {
attrs_.InitDefault();
Init();
}

@@ -78,12 +77,12 @@ void GeModel::SetPlatformVersion(const std::string &platform_version) { this->pl

void GeModel::SetPlatformType(uint8_t platform_type) { this->platform_type_ = platform_type; }

void GeModel::SetAttr(const ProtoAttrMapHelper &attrs) { attrs_ = attrs; }
void GeModel::SetAttr(const ProtoAttrMap &attrs) { attrs_ = attrs; }

ProtoAttrMapHelper GeModel::MutableAttrMap() { return attrs_; }
ProtoAttrMap &GeModel::MutableAttrMap() { return attrs_; }

ConstProtoAttrMapHelper GeModel::GetAttrMap() const {
return ConstProtoAttrMapHelper(attrs_.GetProtoOwner(), attrs_.GetProtoMsg());
ConstProtoAttrMap &GeModel::GetAttrMap() const {
return attrs_;
}

Status GeModel::GetSessionId(uint32_t model_id, uint64_t &session_id) const {


+ 7
- 5
ge/common/model/ge_model.h View File

@@ -17,10 +17,12 @@
#ifndef GE_MODEL_GE_MODEL_H_
#define GE_MODEL_GE_MODEL_H_

#include <securec.h>
#include <map>
#include <memory>
#include <string>

#include "securec.h"
#include "runtime/rt.h"
#include "common/tbe_kernel_store.h"
#include "common/cust_aicpu_kernel_store.h"
#include "framework/common/debug/log.h"
@@ -60,9 +62,9 @@ class GeModel : public AttrHolder {
void SetPlatformVersion(const std::string &platform_version);
void SetPlatformType(uint8_t platform_type);

void SetAttr(const ProtoAttrMapHelper &attrs);
void SetAttr(const ProtoAttrMap &attrs);

ProtoAttrMapHelper MutableAttrMap() override;
ProtoAttrMap &MutableAttrMap() override;

using AttrHolder::SetAttr;
using AttrHolder::GetAllAttrs;
@@ -77,12 +79,12 @@ class GeModel : public AttrHolder {
}

protected:
ConstProtoAttrMapHelper GetAttrMap() const override;
ConstProtoAttrMap &GetAttrMap() const override;

private:
void Init();

ProtoAttrMapHelper attrs_; /*lint !e148*/
ProtoAttrMap attrs_; /*lint !e148*/

Graph graph_;
std::shared_ptr<domi::ModelTaskDef> task_; /*lint !e148*/


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 8b0650bd36d8404b5ad5b6253499734f6504974f
Subproject commit a0004e3459e821b90915238ced08511353834b0c

Loading…
Cancel
Save