1 /** 2 * Copyright 2019-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 #ifndef MINDSPORE_CCSRC_BACKEND_SESSION_SESSION_CONTEXT_H 17 #define MINDSPORE_CCSRC_BACKEND_SESSION_SESSION_CONTEXT_H 18 #include <vector> 19 #include <map> 20 #include <functional> 21 #include <memory> 22 #include <utility> 23 #include <string> 24 25 #include "ir/tensor.h" 26 #include "pipeline/jit/resource_base.h" 27 #include "utils/ms_context.h" 28 namespace mindspore { 29 namespace session { 30 const char kInputCtrlTensors[] = "input_ctrl_tensors"; 31 32 class Context : public pipeline::ResourceBase { 33 public: 34 explicit Context(std::string target = kAscendDevice, uint32_t device_id = 0) target_(std::move (target))35 : target_(std::move(target)), device_id_(device_id) {} 36 ~Context() override = default; 37 device_id()38 uint32_t device_id() const { return device_id_; } 39 static std::shared_ptr<Context> GetInstance(); AddManager(const FuncGraphManagerPtr & m)40 void AddManager(const FuncGraphManagerPtr &m) { manager_list_.push_back(m); } 41 42 private: 43 std::vector<FuncGraphManagerPtr> manager_list_; 44 std::string target_; 45 uint32_t device_id_; 46 }; 47 } // namespace session 48 } // namespace mindspore 49 50 #endif // MINDSPORE_CCSRC_BACKEND_SESSION_SESSION_CONTEXT_H 51