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_test.sh 1.7 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  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 test [OPTIONS]
  21. Options:
  22. -u, --unit Run unit Test
  23. -c, --component Run component Test
  24. -h, --help
  25. EOF
  26. }
  27. function unit_test(){
  28. ${PROJECT_HOME}/build.sh -u
  29. }
  30. function component_test(){
  31. ${PROJECT_HOME}/build.sh -s
  32. }
  33. function parse_args(){
  34. parsed_args=$(getopt -a -o uch --long unit,component,help -- "$@") || {
  35. help
  36. exit 1
  37. }
  38. if [ $# -lt 1 ]; then
  39. unit_test
  40. exit 1
  41. fi
  42. eval set -- "$parsed_args"
  43. while true; do
  44. case "$1" in
  45. -u | --unit)
  46. unit_test
  47. ;;
  48. -c | --component)
  49. component_test
  50. ;;
  51. -h | --help)
  52. help
  53. ;;
  54. --)
  55. shift; break;
  56. ;;
  57. *)
  58. help; exit 1
  59. ;;
  60. esac
  61. shift
  62. done
  63. }
  64. function main(){
  65. parse_args "$@"
  66. }
  67. main "$@"
  68. set +e

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