Hello World 样例
本样例展示了 PyPTO 最简单的张量加法操作,适合初次接触框架的开发者快速了解基本工作流。
总览介绍
本样例通过一个简单的张量加法演示了 PyPTO 的完整开发流程:
- 使用
@pypto.frontend.jit定义内核函数。 - 通过 PyTorch 张量创建输入数据。
- 调用 JIT 内核执行计算并验证结果。
代码结构
hello_world.py: Hello World 示例脚本,包含最基础的张量加法操作。
运行方法
环境准备
# 配置 CANN 环境变量
# 安装完成后请配置环境变量,请用户根据set_env.sh的实际路径执行如下命令。
# 上述环境变量配置只在当前窗口生效,用户可以按需将以上命令写入环境变量配置文件(如.bashrc文件)。
# 默认路径安装,以root用户为例(非root用户,将/usr/local替换为${HOME})
source /usr/local/Ascend/ascend-toolkit/set_env.sh
# 设置设备 ID
export TILE_FWK_DEVICE_ID=0
执行脚本
# 运行 Hello World 示例
python3 hello_world.py
# 以仿真模式运行
python3 hello_world.py --run_mode sim
注意事项
- 本样例是入门的第一步,完成后建议继续学习 初级样例 (01_beginner)。