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

注意事项