From 581a23dae0d24cfa362eb9cb8ad2236cbb6a1cc0 Mon Sep 17 00:00:00 2001
From: ZhouChen <zhouchen53@huawei.com>
Date: Thu, 21 Aug 2025 09:56:39 +0800
Subject: [PATCH] make option aicore_num session level, remove global level
 aicore_num

---
 tf_adapter/util/ge_plugin.cc | 4 ----
 tf_adapter/util/npu_attrs.cc | 6 ------
 2 files changed, 10 deletions(-)

diff --git a/tf_adapter/util/ge_plugin.cc b/tf_adapter/util/ge_plugin.cc
index f9c4f5522..52223fa27 100644
--- a/tf_adapter/util/ge_plugin.cc
+++ b/tf_adapter/util/ge_plugin.cc
@@ -297,10 +297,6 @@ void GePlugin::Init(std::map<std::string, std::string> &init_options, const bool
     ADP_LOG(INFO) << "[GePlugin] export_compile_stat : " << init_options["ge.exportCompileStat"];
   }
 
-  if (init_options.find("ge.aicoreNum") != init_options.end()) {
-    ADP_LOG(INFO) << "[GePlugin] aicoreNum : " << init_options["ge.aicoreNum"];
-  }
-
   if (init_options.find("ge.oo.constantFolding") != init_options.end()) {
     ADP_LOG(INFO) << "[GePlugin] oo_constant_folding : " << init_options["ge.oo.constantFolding"];
   }
diff --git a/tf_adapter/util/npu_attrs.cc b/tf_adapter/util/npu_attrs.cc
index 4741ecc56..f028a69da 100644
--- a/tf_adapter/util/npu_attrs.cc
+++ b/tf_adapter/util/npu_attrs.cc
@@ -708,7 +708,6 @@ std::map<std::string, std::string> NpuAttrs::GetInitOptions(const OpKernelConstr
   std::string stream_sync_timeout = "-1";
   std::string event_sync_timeout = "-1";
   std::string export_compile_stat;
-  std::string aicore_num;
   std::string oo_constant_folding;
   std::string input_batch_cpy;
 
@@ -752,7 +751,6 @@ std::map<std::string, std::string> NpuAttrs::GetInitOptions(const OpKernelConstr
     (void) ctx->GetAttr("_stream_sync_timeout", &stream_sync_timeout);
     (void) ctx->GetAttr("_event_sync_timeout", &event_sync_timeout);
     (void) ctx->GetAttr("_export_compile_stat", &export_compile_stat);
-    (void) ctx->GetAttr("_aicore_num", &aicore_num);
     (void) ctx->GetAttr("_oo_constant_folding", &oo_constant_folding);
     (void) ctx->GetAttr("_input_batch_cpy", &input_batch_cpy);
   }
@@ -814,8 +812,6 @@ std::map<std::string, std::string> NpuAttrs::GetInitOptions(const OpKernelConstr
     init_options_["export_compile_stat"] = export_compile_stat;
     init_options_["ge.exportCompileStat"] = export_compile_stat;
   }
-  init_options_["aicore_num"] = aicore_num;
-  init_options_["ge.aicoreNum"] = aicore_num;
   if (!oo_constant_folding.empty()) {
     init_options_["oo_constant_folding"] = oo_constant_folding;
     init_options_["ge.oo.constantFolding"] = oo_constant_folding;
@@ -2493,8 +2489,6 @@ Status NpuAttrs::SetNpuOptimizerAttr(const GraphOptimizationPassOptions &options
       }
       if ((params.count("aicore_num") > 0)) {
         aicore_num = params.at("aicore_num").s();
-        init_options_["aicore_num"] = aicore_num;
-        init_options_["ge.aicoreNum"] = aicore_num;
       }
       // input_batch_cpy
       if (params.count("input_batch_cpy") > 0) {
-- 
Gitee