diff --git a/include/api/context.h b/include/api/context.h index d1335ff5e67..69726ee59b1 100644 --- a/include/api/context.h +++ b/include/api/context.h @@ -147,7 +147,7 @@ class MS_API Context { /// heterogeneous scenarios with multiple members in the vector. /// /// \return Mutable reference of DeviceInfoContext vector in this context. - std::vector> &MutableDeviceInfo() const; + std::vector> &MutableDeviceInfo(); private: std::shared_ptr data_; diff --git a/mindspore/ccsrc/cxx_api/context.cc b/mindspore/ccsrc/cxx_api/context.cc index 5bbf7352cb5..bc49adbf338 100644 --- a/mindspore/ccsrc/cxx_api/context.cc +++ b/mindspore/ccsrc/cxx_api/context.cc @@ -95,7 +95,7 @@ std::vector Context::GetThreadAffinityCoreList() const { return data_->affinity_core_list_; } -std::vector> &Context::MutableDeviceInfo() const { +std::vector> &Context::MutableDeviceInfo() { MS_EXCEPTION_IF_NULL(data_); return data_->device_info_list; } diff --git a/mindspore/lite/src/extendrt/cxx_api/context.cc b/mindspore/lite/src/extendrt/cxx_api/context.cc index b309c45f73b..a22aea212a4 100644 --- a/mindspore/lite/src/extendrt/cxx_api/context.cc +++ b/mindspore/lite/src/extendrt/cxx_api/context.cc @@ -211,7 +211,7 @@ bool Context::GetMultiModalHW() const { return data_->float_mode; } -std::vector> &Context::MutableDeviceInfo() const { +std::vector> &Context::MutableDeviceInfo() { static std::vector> empty{}; if (data_ == nullptr) { MS_LOG(ERROR) << "Invalid context."; diff --git a/mindspore/lite/src/litert/cxx_api/context.cc b/mindspore/lite/src/litert/cxx_api/context.cc index b931faeee59..e149168d48f 100644 --- a/mindspore/lite/src/litert/cxx_api/context.cc +++ b/mindspore/lite/src/litert/cxx_api/context.cc @@ -231,7 +231,7 @@ bool Context::GetMultiModalHW() const { return data_->float_mode; } -std::vector> &Context::MutableDeviceInfo() const { +std::vector> &Context::MutableDeviceInfo() { static std::vector> empty{}; if (data_ == nullptr) { MS_LOG(ERROR) << "Invalid context.";