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.

ge_clean.sh 2.2 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. #!/bin/bash
  2. # Copyright 2021 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. set -e
  17. PROJECT_HOME=${PROJECT_HOME:-$(dirname "$0")/../../}
  18. function help(){
  19. cat <<-EOF
  20. Usage: ge clean [OPTIONS]
  21. Options:
  22. -b, --build Clean build dir
  23. -d, --docs Clean generate docs
  24. -i, --install Clean dependenices
  25. -a, --all Clean all
  26. -h, --help
  27. EOF
  28. }
  29. function clean_relative_dir(){
  30. rm -rf "${PROJECT_HOME}/${1:-output}"
  31. }
  32. function parse_args(){
  33. parsed_args=$(getopt -a -o bdiah --long build,docs,install,all,help -- "$@") || {
  34. help
  35. exit 1
  36. }
  37. if [ $# -lt 1 ]; then
  38. clean_relative_dir "build"
  39. clean_relative_dir "output"
  40. exit 1
  41. fi
  42. eval set -- "$parsed_args"
  43. while true; do
  44. case "$1" in
  45. -b | --build)
  46. clean_relative_dir "build"
  47. clean_relative_dir "output"
  48. ;;
  49. -d | --docs)
  50. clean_relative_dir "docs/doxygen"
  51. ;;
  52. -i | --install)
  53. clean_relative_dir "deps"
  54. ;;
  55. -a | --all)
  56. clean_relative_dir "deps"
  57. clean_relative_dir "build"
  58. clean_relative_dir "output"
  59. clean_relative_dir "docs/doxygen"
  60. ;;
  61. -h | --help)
  62. help
  63. ;;
  64. --)
  65. shift; break;
  66. ;;
  67. *)
  68. help; exit 1
  69. ;;
  70. esac
  71. shift
  72. done
  73. }
  74. function main(){
  75. parse_args "$@"
  76. }
  77. main "$@"
  78. set +e

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