Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 7f538b51e7
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7f538b51e7'
${ noResults }
mindspore/mindspore/ccsrc/utils/context/context_extends.h

52 lines
2.0 kB
Raw Blame History 

  1. /**

  2.  * Copyright 2019 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. #ifndef MINDSPORE_CCSRC_UTILS_CONTEXT_CONTEXT_EXTENDS_H

  18. #define MINDSPORE_CCSRC_UTILS_CONTEXT_CONTEXT_EXTENDS_H

  19. 

  20. #include <map>

  21. #include <string>

  22. #include <memory>

  23. #include "utils/ms_context.h"

  24. #include "utils/tensorprint_utils.h"

  25. 

  26. #ifndef NO_DLIB

  27. #include "tdt/tsd_client.h"

  28. #include "tdt/tdt_host_interface.h"

  29. #include "tdt/data_common.h"

  30. #include "runtime/dev.h"

  31. #endif

  32. #ifdef ENABLE_GE

  33. #include "transform/graph_ir/df_graph_manager.h"

  34. #endif

  35. 

  36. namespace mindspore {

  37. namespace context {

  38. bool OpenTsd(const std::shared_ptr<MsContext> &inst_context);

  39. bool CloseTsd(const std::shared_ptr<MsContext> &inst_context, bool force = false);

  40. void SetHcclOptions(const std::shared_ptr<MsContext> &inst_context, std::map<std::string, std::string> *ge_options);

  41. void GetGeOptions(const std::shared_ptr<MsContext> &inst_context, std::map<std::string, std::string> *ge_options);

  42. void SetDisableReuseMemoryFlag(std::map<std::string, std::string> *ge_options);

  43. bool InitGe(const std::shared_ptr<MsContext> &inst_context);

  44. bool FinalizeGe(const std::shared_ptr<MsContext> &inst_context, bool force = false);

  45. bool PynativeInitGe(const std::shared_ptr<MsContext> &inst_context);

  46. bool IsTsdOpened(const std::shared_ptr<MsContext> &inst_context);

  47. bool IsGeInited(const std::shared_ptr<MsContext> &inst_context);

  48. }  // namespace context

  49. }  // namespace mindspore

  50. 

  51. #endif  // MINDSPORE_CCSRC_UTILS_CONTEXT_CONTEXT_EXTENDS_H