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.

hybrid_davinci_model_stub.cc 3.2 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  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. #include "hybrid_davinci_model.h"
  17. namespace ge {
  18. namespace hybrid {
  19. HybridDavinciModel::~HybridDavinciModel() {}
  20. std::unique_ptr<HybridDavinciModel> HybridDavinciModel::Create(const GeRootModelPtr &ge_root_model) {
  21. return std::unique_ptr<HybridDavinciModel>(new (std::nothrow)HybridDavinciModel());
  22. }
  23. Status HybridDavinciModel::Init() {
  24. return UNSUPPORTED;
  25. }
  26. Status HybridDavinciModel::Execute(const std::vector<DataBuffer> &inputs,
  27. const std::vector<GeTensorDesc> &input_desc,
  28. std::vector<DataBuffer> &outputs,
  29. std::vector<GeTensorDesc> &output_desc,
  30. rtStream_t stream) {
  31. return UNSUPPORTED;
  32. }
  33. Status HybridDavinciModel::Execute(const vector<GeTensor> &inputs, vector<GeTensor> &outputs) {
  34. return UNSUPPORTED;
  35. }
  36. Status HybridDavinciModel::ModelRunStart() {
  37. return UNSUPPORTED;
  38. }
  39. Status HybridDavinciModel::ModelRunStop() {
  40. return UNSUPPORTED;
  41. }
  42. Status HybridDavinciModel::EnqueueData(const shared_ptr<InputDataWrapper> &data) {
  43. return UNSUPPORTED;
  44. }
  45. void HybridDavinciModel::SetListener(const shared_ptr<ModelListener> &listener) {
  46. }
  47. void HybridDavinciModel::SetModelId(uint32_t model_id) {
  48. }
  49. void HybridDavinciModel::SetDeviceId(uint32_t device_id) {
  50. }
  51. void HybridDavinciModel::SetOmName(const string &om_name) {
  52. }
  53. uint64_t HybridDavinciModel::GetSessionId() {
  54. return 0;
  55. }
  56. uint32_t HybridDavinciModel::GetDataInputerSize() {
  57. return 0;
  58. }
  59. Status HybridDavinciModel::GetDynamicBatchInfo(std::vector<std::vector<int64_t>> &batch_info, int32_t &dynamic_type) {
  60. return UNSUPPORTED;
  61. }
  62. void HybridDavinciModel::GetUserDesignateShapeOrder(std::vector<std::string> &user_input_shape_order) {
  63. }
  64. void HybridDavinciModel::GetModelAttr(std::vector<std::string> &dynamic_output_shape_info) {
  65. }
  66. Status HybridDavinciModel::GetInputOutputDescInfo(vector<InputOutputDescInfo> &input_desc,
  67. vector<InputOutputDescInfo> &output_desc,
  68. std::vector<uint32_t> &input_formats,
  69. std::vector<uint32_t> &output_formats) {
  70. return UNSUPPORTED;
  71. }
  72. void HybridDavinciModel::SetModelDescVersion(bool is_new_model_desc) {
  73. }
  74. bool HybridDavinciModel::GetRunningFlag() const {
  75. return false;
  76. }
  77. Status HybridDavinciModel::SetRunAsyncListenerCallback(const RunAsyncCallback &callback) {
  78. return UNSUPPORTED;
  79. }
  80. } // namespace hybrid
  81. } // namespace ge

图引擎模块(GE)是MindSpore的一个子模块,其代码由C++实现,位于前端模块ME和底层硬件之间,起到承接作用。图引擎模块以ME下发的图作为输入,然后进行一系列的深度图优化操作,最后输出一张可以在底层硬件上高效运行的图。GE针对昇腾AI处理器的硬件结构特点,做了特定的优化工作,以此来充分发挥出昇腾AI处理器的强大算力。在进行模型训练/推理时,GE会被自动调用而用户并不感知。GE主要由GE API和GE Core两部分组成,详细的架构图如下所示