1 /** 2 * Copyright 2020 Huawei Technologies Co., Ltd 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #include "ps/scheduler.h" 18 19 namespace mindspore { 20 namespace ps { Run()21void Scheduler::Run() { 22 MS_LOG(INFO) << "Start scheduler."; 23 PSContext::instance()->cluster_config().scheduler_host = PSContext::instance()->scheduler_host(); 24 PSContext::instance()->cluster_config().scheduler_port = PSContext::instance()->scheduler_port(); 25 PSContext::instance()->cluster_config().initial_worker_num = PSContext::instance()->initial_worker_num(); 26 PSContext::instance()->cluster_config().initial_server_num = PSContext::instance()->initial_server_num(); 27 if (!scheduler_node_.Start()) { 28 MS_LOG(WARNING) << "Scheduler start failed."; 29 } 30 31 if (!scheduler_node_.Finish()) { 32 MS_LOG(WARNING) << "Scheduler finis failed."; 33 } 34 35 if (!scheduler_node_.Stop()) { 36 MS_LOG(WARNING) << "Scheduler stop failed."; 37 } 38 exit(1); 39 } 40 } // namespace ps 41 } // namespace mindspore 42