From 38e4de4c171ab375be3da5df85952830722f702b Mon Sep 17 00:00:00 2001
From: guopeian <guopeian1@hisilicon.com>
Date: Tue, 30 Mar 2021 10:04:32 +0800
Subject: [PATCH] add hong

---
 ge/plugin/engine/dnnengines.cc    | 24 ++++++++++++++++++++----
 ge/plugin/engine/engine_manage.cc |  8 ++++++++
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/ge/plugin/engine/dnnengines.cc b/ge/plugin/engine/dnnengines.cc
index 5b06310c..f69cf769 100755
--- a/ge/plugin/engine/dnnengines.cc
+++ b/ge/plugin/engine/dnnengines.cc
@@ -52,7 +52,7 @@ Status VectorCoreDNNEngine::Initialize(const std::map<std::string, std::string>
 Status VectorCoreDNNEngine::Finalize() { return SUCCESS; }
 
 void VectorCoreDNNEngine::GetAttributes(DNNEngineAttribute &attrs) const { attrs = engine_attribute_; }
-
+#ifndef ONLY_COMPILE_OPEN_SRC
 AICpuDNNEngine::AICpuDNNEngine(const std::string &engine_name) {
   engine_attribute_.engine_name = engine_name;
   engine_attribute_.compute_cost = COST_2;
@@ -60,7 +60,15 @@ AICpuDNNEngine::AICpuDNNEngine(const std::string &engine_name) {
   engine_attribute_.engine_input_format = FORMAT_RESERVED;
   engine_attribute_.engine_output_format = FORMAT_RESERVED;
 }
-
+#else
+AICpuDNNEngine::AICpuDNNEngine(const std::string &engine_name) {
+  engine_attribute_.engine_name = engine_name;
+  engine_attribute_.compute_cost = COST_3;
+  engine_attribute_.runtime_type = DEVICE;
+  engine_attribute_.engine_input_format = FORMAT_RESERVED;
+  engine_attribute_.engine_output_format = FORMAT_RESERVED;
+}
+#endif
 AICpuDNNEngine::AICpuDNNEngine(const DNNEngineAttribute &attrs) {  engine_attribute_ = attrs; }
 
 Status AICpuDNNEngine::Initialize(const std::map<std::string, std::string> &options) { return SUCCESS; }
@@ -68,7 +76,7 @@ Status AICpuDNNEngine::Initialize(const std::map<std::string, std::string> &opti
 Status AICpuDNNEngine::Finalize() { return SUCCESS; }
 
 void AICpuDNNEngine::GetAttributes(DNNEngineAttribute &attrs) const { attrs = engine_attribute_; }
-
+#ifndef ONLY_COMPILE_OPEN_SRC
 AICpuTFDNNEngine::AICpuTFDNNEngine(const std::string &engine_name) {
   engine_attribute_.engine_name = engine_name;
   engine_attribute_.compute_cost = COST_3;
@@ -76,7 +84,15 @@ AICpuTFDNNEngine::AICpuTFDNNEngine(const std::string &engine_name) {
   engine_attribute_.engine_input_format = FORMAT_RESERVED;
   engine_attribute_.engine_output_format = FORMAT_RESERVED;
 }
-
+#else
+AICpuTFDNNEngine::AICpuTFDNNEngine(const std::string &engine_name) {
+  engine_attribute_.engine_name = engine_name;
+  engine_attribute_.compute_cost = COST_2;
+  engine_attribute_.runtime_type = DEVICE;
+  engine_attribute_.engine_input_format = FORMAT_RESERVED;
+  engine_attribute_.engine_output_format = FORMAT_RESERVED;
+}
+#endif
 AICpuTFDNNEngine::AICpuTFDNNEngine(const DNNEngineAttribute &attrs) {  engine_attribute_ = attrs; }
 
 Status AICpuTFDNNEngine::Initialize(const std::map<std::string, std::string> &options) { return SUCCESS; }
diff --git a/ge/plugin/engine/engine_manage.cc b/ge/plugin/engine/engine_manage.cc
index 11fdfe89..0fdfb672 100644
--- a/ge/plugin/engine/engine_manage.cc
+++ b/ge/plugin/engine/engine_manage.cc
@@ -92,7 +92,11 @@ void RegisterAiCpuEngine() {
   const std::string vm_aicpu = "DNN_VM_AICPU_ASCEND";
   std::vector<std::string> mem_type_aicpu;
   mem_type_aicpu.emplace_back(GE_ENGINE_ATTR_MEM_TYPE_HBM);
+#ifndef ONLY_COMPILE_OPEN_SRC
   DNNEngineAttribute attr_aicpu = {vm_aicpu, mem_type_aicpu, COST_2, DEVICE, FORMAT_RESERVED, FORMAT_RESERVED};
+#else
+  DNNEngineAttribute attr_aicpu = {vm_aicpu, mem_type_aicpu, COST_3, DEVICE, FORMAT_RESERVED, FORMAT_RESERVED};
+#endif
   DNNEnginePtr vm_engine_ptr = MakeShared<AICpuDNNEngine>(attr_aicpu);
   if (vm_engine_ptr == nullptr) {
     GELOGE(ge::FAILED, "make vm_engine_ptr failed");
@@ -107,7 +111,11 @@ void RegisterAiCpuTFEngine() {
   const std::string vm_aicpu_tf = "DNN_VM_AICPU";
   std::vector<std::string> mem_type_aicpu_tf;
   mem_type_aicpu_tf.emplace_back(GE_ENGINE_ATTR_MEM_TYPE_HBM);
+#ifndef ONLY_COMPILE_OPEN_SRC
   DNNEngineAttribute attr_aicpu_tf = {vm_aicpu_tf, mem_type_aicpu_tf, COST_3, DEVICE, FORMAT_RESERVED, FORMAT_RESERVED};
+#else
+  DNNEngineAttribute attr_aicpu_tf = {vm_aicpu_tf, mem_type_aicpu_tf, COST_2, DEVICE, FORMAT_RESERVED, FORMAT_RESERVED};
+#endif
   DNNEnginePtr vm_engine_ptr = MakeShared<AICpuTFDNNEngine>(attr_aicpu_tf);
   if (vm_engine_ptr == nullptr) {
     GELOGE(ge::FAILED, "make vm_engine_ptr failed");