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 2.8 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
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  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. uint64_t HybridDavinciModel::GetSessionId() {
  52. return 0;
  53. }
  54. Status HybridDavinciModel::GetDynamicBatchInfo(std::vector<std::vector<int64_t>> &batch_info, int32_t &dynamic_type) {
  55. return UNSUPPORTED;
  56. }
  57. void HybridDavinciModel::GetUserDesignateShapeOrder(std::vector<std::string> &user_input_shape_order) {
  58. }
  59. void HybridDavinciModel::GetModelAttr(std::vector<std::string> &dynamic_output_shape_info) {
  60. }
  61. Status HybridDavinciModel::GetInputOutputDescInfo(vector<InputOutputDescInfo> &input_desc,
  62. vector<InputOutputDescInfo> &output_desc,
  63. std::vector<uint32_t> &input_formats,
  64. std::vector<uint32_t> &output_formats) {
  65. return UNSUPPORTED;
  66. }
  67. void HybridDavinciModel::SetModelDescVersion(bool is_new_model_desc) {
  68. }
  69. } // namespace hybrid
  70. } // namespace ge

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