CANN/pto-isa核心类型详解

核心类型详解

【免费下载链接】pto-isaParallel Tile Operation (PTO) is a virtual instruction set architecture designed by Ascend CANN, focusing on tile-level operations. This repository offers high-performance, cross-platform tile operations across Ascend platforms.项目地址: https://gitcode.com/cann/pto-isa

Signal — 标量信号

用于单标志同步，封装int32_t类型的 GM 地址：

using Signal = GlobalTensor<int32_t, Shape<1,1,1,1,1>, Stride<1,1,1,1,1>, Layout::ND>; comm::Signal sig(ptr); // ptr: __gm__ int32_t*

Signal2D — 二维信号矩阵

编译期形状的二维信号网格，支持密集布局和子区域视图：

// 密集 4×8 网格（步长自动推导为 8） comm::Signal2D<4, 8> grid(ptr); // 从 128 列大网格中的子区域（步长 = 128） comm::Signal2D<4, 8> sub(ptr + offset, 128);

ParallelGroup — 集合通信分组

轻量级视图，封装多 rank 的GlobalTensor对象数组：

template <typename GlobalData> struct ParallelGroup { GlobalData *tensors; // 每个 rank 的 GlobalTensor 数组 int nranks; // rank 总数 int rootIdx; // root NPU 的 rank 索引 static ParallelGroup Create(GlobalData *tensorArray, int size, int rootIdx); };

关键约束：

• tensors指向外部数组（不做动态内存分配）
• rootIdx是 root rank 在组中的索引，所有 rank 必须传入相同的rootIdx
• 通过operator[]按 team rank 索引访问

NotifyOp — 通知操作类型

WaitCmp — 比较运算符

ReduceOp — 归约运算符

AtomicType — 原子操作类型

定义于include/pto/common/constants.hpp：

DmaEngine — DMA 引擎选择

AsyncEvent — 异步事件句柄

struct AsyncEvent { uint64_t handle; DmaEngine engine; bool valid() const; // handle != 0 时返回 true bool Wait(const AsyncSession &session) const; // 阻塞直到传输完成 bool Test(const AsyncSession &session) const; // 非阻塞完成检测 };

AsyncSession — 异步会话

引擎无关的会话对象，通过BuildAsyncSession<engine>()构建：

struct AsyncSession { DmaEngine engine; sdma::SdmaSession sdmaSession; urma::UrmaSession urmaSession; bool valid; };

【免费下载链接】pto-isaParallel Tile Operation (PTO) is a virtual instruction set architecture designed by Ascend CANN, focusing on tile-level operations. This repository offers high-performance, cross-platform tile operations across Ascend platforms.项目地址: https://gitcode.com/cann/pto-isa