From 67b1ab0c7dd93b0decbd5c9aa0962747b0b7fcbf Mon Sep 17 00:00:00 2001 From: xiaguozheng Date: Thu, 30 Oct 2025 19:31:51 +0800 Subject: [PATCH] ge.inputPlacement=DeviceHbm --- tf_adapter/util/npu_attrs.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tf_adapter/util/npu_attrs.cc b/tf_adapter/util/npu_attrs.cc index 76db01955..80f512b66 100644 --- a/tf_adapter/util/npu_attrs.cc +++ b/tf_adapter/util/npu_attrs.cc @@ -661,6 +661,7 @@ std::map NpuAttrs::GetSessOptions(const OpKernelConstr } sess_options["ge.inputBatchCpy"] = input_batch_cpy; sess_options["input_batch_cpy"] = input_batch_cpy; + sess_options["ge.inputPlacement"] = "DeviceHbm"; sess_options[ge::OPTION_ALL_TENSOR_NOT_EMPTY] = all_tensor_not_empty; sess_options["all_tensor_not_empty"] = all_tensor_not_empty; sess_options["auto_multistream_parallel_mode"] = auto_multistream_parallel_mode; @@ -842,6 +843,7 @@ std::map NpuAttrs::GetInitOptions(const OpKernelConstr } init_options_["input_batch_cpy"] = input_batch_cpy; init_options_["ge.inputBatchCpy"] = input_batch_cpy; + init_options_["ge.inputPlacement"] = "DeviceHbm"; SetForbiddenClosePassOn(init_options_); return init_options_; } @@ -1837,6 +1839,7 @@ std::map NpuAttrs::GetAllAttrOptions(const AttrSlice & all_options["ge.oo.constantFolding"] = oo_constant_folding; } // input_batch_cpy + all_options["ge.inputPlacement"] = "DeviceHbm"; all_options["input_batch_cpy"] = input_batch_cpy; all_options["ge.inputBatchCpy"] = input_batch_cpy; all_options["shape_generalization_mode"] = shape_generalization_mode; @@ -2565,6 +2568,7 @@ Status NpuAttrs::SetNpuOptimizerAttr(const GraphOptimizationPassOptions &options const auto input_batch_cpy_str = input_batch_cpy ? "true" : "false"; init_options_["input_batch_cpy"] = input_batch_cpy_str; init_options_["ge.inputBatchCpy"] = input_batch_cpy_str; + init_options_["ge.inputPlacement"] = "DeviceHbm"; } if (params.count("jit_compile") > 0) { const static std::vector kJitCompileList = {"true", @@ -2653,6 +2657,7 @@ Status NpuAttrs::SetNpuOptimizerAttr(const GraphOptimizationPassOptions &options sess_options["input_fusion_size"] = std::to_string(input_fusion_size); sess_options["input_batch_cpy"] = std::to_string(input_batch_cpy); sess_options["ge.inputBatchCpy"] = std::to_string(input_batch_cpy); + sess_options["ge.inputPlacement"] = "DeviceHbm"; sess_options["aicore_num"] = aicore_num; sess_options["ge.aicoreNum"] = aicore_num; sess_options["all_tensor_not_empty"] = std::to_string(all_tensor_not_empty); -- Gitee