From 516938c95e1fd9cf8b149078214ba10daf7ed318 Mon Sep 17 00:00:00 2001 From: Bellatan Date: Sat, 15 Mar 2025 11:17:04 +0000 Subject: [PATCH] 1.OH_AI_TensorClone api set_own_data true 2.sync mstendor and lite_tensor own_data Signed-off-by: Bellatan --- .../source/mindspore/lite/src/litert/c_api/tensor_c.cc | 2 +- .../mindspore/lite/src/litert/cxx_api/tensor/tensor_impl.h | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore-src/source/mindspore/lite/src/litert/c_api/tensor_c.cc b/mindspore-src/source/mindspore/lite/src/litert/c_api/tensor_c.cc index 63096fa255..95c2991283 100644 --- a/mindspore-src/source/mindspore/lite/src/litert/c_api/tensor_c.cc +++ b/mindspore-src/source/mindspore/lite/src/litert/c_api/tensor_c.cc @@ -91,7 +91,7 @@ OH_AI_TensorHandle OH_AI_TensorClone(OH_AI_TensorHandle tensor) { MS_LOG(ERROR) << "Failed to allocate tensor impl."; return nullptr; } - std::static_pointer_cast(clone_impl->impl())->set_own_data(false); + std::static_pointer_cast(clone_impl->impl())->set_own_data(true); clone_impl->SetTensorName(impl->Name() + "_duplicate"); return clone_impl; } diff --git a/mindspore-src/source/mindspore/lite/src/litert/cxx_api/tensor/tensor_impl.h b/mindspore-src/source/mindspore/lite/src/litert/cxx_api/tensor/tensor_impl.h index cfddacfb2e..410b98c463 100644 --- a/mindspore-src/source/mindspore/lite/src/litert/cxx_api/tensor/tensor_impl.h +++ b/mindspore-src/source/mindspore/lite/src/litert/cxx_api/tensor/tensor_impl.h @@ -294,7 +294,9 @@ class LiteTensorImpl : public MutableTensorImpl { return kSuccess; } - void set_own_data(bool own_data) { own_data_ = own_data; } + void set_own_data(bool own_data) { + lite_tensor_->set_own_data(own_data); + own_data_ = own_data; } void set_from_session(bool from_session) { from_session_ = from_session; } -- Gitee