样例使用指导

1、功能描述

本样例使用Sigmod私有属性进行构图,旨在帮助构图开发者快速理解私有属性的定义并使用私有属性进行构图

2、目录结构

cpp/
├── src/
|   └── CMakeLists.txt           // 编译脚本
|   └── es_showcase.h            // 头文件
|   └── make_sigmod_graph.cpp    // sample文件
├── CMakeLists.txt               // 编译脚本
├── main.cpp                     // 程序主入口
├── README.md                    // README文件
├── run_sample.sh                // 执行脚本
├── utils.h                      // 工具文件

3、使用方法

3.1、准备cann包

  • 通过安装指导 环境准备正确安装toolkit
  • 设置环境变量 (假设包安装在/usr/local/Ascend/)
source /usr/local/Ascend/cann/set_env.sh 
  • 通过安装指导正确安装算子ops包 (ES依赖算子原型进行API生成), 并正确配置环境变量
  • 安装算子ops包 (ES依赖算子原型进行API生成)

3.2、编译和执行

只需运行下述命令即可完成清理、生成接口、构图和DUMP图:

bash run_sample.sh

当前 run_sample.sh 的行为是:先自动清理旧的 build,构建 sample并默认执行sample dump 。当看到如下信息,代表执行成功:

[Success] sample 执行成功,pbtxt dump 已生成在当前目录。该文件以 ge_onnx_ 开头,可以在 netron 中打开显示

输出文件说明

执行成功后会在当前目录生成以下文件:

  • ge_onnx_*.pbtxt - 图结构的protobuf文本格式,可用netron查看
  • ge_proto_*.txt - 图结构的txt文本格式,可直接使用文本编辑器查看是否包含"node_rank"属性字段

3.3、日志打印

可执行程序执行过程中如果需要日志打印来辅助定位,可以在bash run_sample.sh之前设置如下环境变量来让日志打印到屏幕

export ASCEND_SLOG_PRINT_TO_STDOUT=1 #日志打印到屏幕
export ASCEND_GLOBAL_LOG_LEVEL=0 #日志级别为debug级别

4、核心概念介绍

4.1、构图步骤如下:

  • 创建图构建器(用于提供构图所需的上下文、工作空间及构建相关方法)
  • 添加起始节点(起始节点指无输入依赖的节点,通常包括图的输入(如 Data 节点)和权重常量(如 Const 节点))
  • 添加中间节点(中间节点为具有输入依赖的计算节点,通常由用户构图逻辑生成,并通过已有节点作为输入连接)
  • 设置图输出(明确图的输出节点,作为计算结果的终点)

4.2、概念说明:

私有属性指的是自定义在ir中不存在的属性,es构图提供对应接口来完成自定义属性的设置