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.

ascend_string.cc 2.6 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. /**
  2. * Copyright 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 "external/graph/ascend_string.h"
  17. #include "debug/ge_log.h"
  18. namespace ge {
  19. AscendString::AscendString(const char* name) {
  20. if (name != nullptr) {
  21. name_ = std::shared_ptr<std::string>(new (std::nothrow) std::string(name)); //lint !e1524
  22. if (name_ == nullptr) {
  23. GELOGE(FAILED, "AscendString[%s] make shared failed.", name);
  24. }
  25. }
  26. }
  27. const char* AscendString::GetString() const {
  28. if (name_ == nullptr) {
  29. return nullptr;
  30. }
  31. return (*name_).c_str();
  32. }
  33. bool AscendString::operator<(const AscendString& d) const {
  34. if (name_ == nullptr && d.name_ == nullptr) {
  35. return false;
  36. } else if (name_ == nullptr) {
  37. return true;
  38. } else if (d.name_ == nullptr) {
  39. return false;
  40. }
  41. return (*name_ < *(d.name_));
  42. }
  43. bool AscendString::operator>(const AscendString& d) const {
  44. if (name_ == nullptr && d.name_ == nullptr) {
  45. return false;
  46. } else if (name_ == nullptr) {
  47. return false;
  48. } else if (d.name_ == nullptr) {
  49. return true;
  50. }
  51. return(*name_ > *(d.name_));
  52. }
  53. bool AscendString::operator==(const AscendString& d) const {
  54. if (name_ == nullptr && d.name_ == nullptr) {
  55. return true;
  56. } else if (name_ == nullptr) {
  57. return false;
  58. } else if (d.name_ == nullptr) {
  59. return false;
  60. }
  61. return (*name_ == *(d.name_));
  62. }
  63. bool AscendString::operator<=(const AscendString& d) const {
  64. if (name_ == nullptr) {
  65. return true;
  66. } else if (d.name_ == nullptr) {
  67. return false;
  68. }
  69. return (*name_ <= *(d.name_));
  70. }
  71. bool AscendString::operator>=(const AscendString& d) const {
  72. if (d.name_ == nullptr) {
  73. return true;
  74. } else if (name_ == nullptr) {
  75. return false;
  76. }
  77. return (*name_ >= *(d.name_));
  78. }
  79. bool AscendString::operator!=(const AscendString& d) const {
  80. if (name_ == nullptr && d.name_ == nullptr) {
  81. return false;
  82. } else if (name_ == nullptr) {
  83. return true;
  84. } else if (d.name_ == nullptr) {
  85. return true;
  86. }
  87. return (*name_ != *(d.name_));
  88. }
  89. } // namespace ge

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