1 /** 2 * Copyright 2019-2023 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/ps/resource_base.h" 27 #include "utils/ms_context.h" 28 namespace mindspore { 29 namespace session { 30 class Context : public pipeline::ResourceBase { 31 public: 32 explicit Context(std::string target = kAscendDevice, uint32_t device_id = 0) target_(std::move (target))33 : target_(std::move(target)), device_id_(device_id) {} 34 ~Context() override = default; 35 device_id()36 uint32_t device_id() const { return device_id_; } AddManager(const FuncGraphManagerPtr & m)37 void AddManager(const FuncGraphManagerPtr &m) { manager_list_.push_back(m); } 38 39 private: 40 std::vector<FuncGraphManagerPtr> manager_list_; 41 std::string target_; 42 uint32_t device_id_; 43 }; 44 } // namespace session 45 } // namespace mindspore 46 47 #endif // MINDSPORE_CCSRC_BACKEND_SESSION_SESSION_CONTEXT_H 48