Browse Source

!42 update mindspore to 1210

Merge pull request !42 from 徐永飞/update_mindspore
tags/v1.1.0
zhangyinxia Gitee 5 years ago
parent
commit
e3c754e906
3 changed files with 13 additions and 2 deletions
  1. +9
    -1
      mindspore_serving/ccsrc/worker/worker.cc
  2. +3
    -0
      mindspore_serving/ccsrc/worker/worker.h
  3. +1
    -1
      third_party/mindspore

+ 9
- 1
mindspore_serving/ccsrc/worker/worker.cc View File

@@ -75,7 +75,7 @@ Status Worker::RegisterWorker() {
}

Status Worker::StartVersionController() {
version_controller_.StartPollModelPeriodic();
// first disable auto updated
return SUCCESS;
}

@@ -110,6 +110,13 @@ Status Worker::RemoveWorker(const ServableWorkerContext &work) {
}

Status Worker::Run(const proto::PredictRequest &request, proto::PredictReply *reply) {
std::shared_lock<std::shared_mutex> lock(worker_shared_lock_);
if (servable_stoppedd_) {
return INFER_STATUS_LOG_ERROR(FAILED) << "Run worker for inference failed, worker has been stopped";
}
if (!servable_started_) {
return INFER_STATUS_LOG_ERROR(FAILED) << "Run worker for inference failed, worker has not been started";
}
MSI_EXCEPTION_IF_NULL(reply);
std::vector<InstanceData> inputs;
RequestSpec request_spec;
@@ -347,6 +354,7 @@ void Worker::StopServable(bool notify_master) {
}

void Worker::Clear() {
std::unique_lock<std::shared_mutex> lock(worker_shared_lock_);
if (clear_flag_.test_and_set()) {
return;
}


+ 3
- 0
mindspore_serving/ccsrc/worker/worker.h View File

@@ -22,6 +22,7 @@
#include <vector>
#include <string>
#include <utility>
#include <shared_mutex>
#include "worker/work_executor.h"
#include "common/serving_common.h"
#include "proto/ms_service.pb.h"
@@ -108,6 +109,8 @@ class MS_API Worker {
std::atomic_flag clear_flag_ = ATOMIC_FLAG_INIT;
std::shared_ptr<BaseNotifyMaster> notify_master_ = nullptr;

std::shared_mutex worker_shared_lock_;

Status LoadServableConfig(const LoadServableSpec &servable_spec, const std::string &version_strategy,
std::vector<uint64_t> *real_version_number);
void GetVersions(const LoadServableSpec &servable_spec, std::vector<uint64_t> *real_versions);


+ 1
- 1
third_party/mindspore

@@ -1 +1 @@
Subproject commit 2403618aaa00d731f10e907ff12388abc3f1c751
Subproject commit defb644c72edcd161e6011ef6d5dd92a51dbb802

Loading…
Cancel
Save