pypto_pro.language.STPhase
产品支持情况
- Ascend 950PR&950DT系列产品:支持
- Atlas A3系列产品:不支持
- Atlas A2系列产品:不支持
功能说明
矩阵计算结果从L0C Buffer分阶段搬出时使用的阶段枚举。用于store、store_tile、move和insert接口,与矩阵计算接口的pypto_pro.language.AccPhase配合,保证L0C Buffer中的结果在计算完成后搬往GM、UB或L1 Buffer,并在最终搬出后允许后续计算复用该存储空间。
对应的matmul、matmul_acc、matmul_mx或matmul_mx_acc配置AccPhase时,搬出结果的store、store_tile、move或insert也必须配置STPhase。矩阵计算的最后一次写操作须使用AccPhase.Final,计算结果的最后一次搬出须使用STPhase.Final。对同一计算结果执行多次搬出时,最后一次之前使用STPhase.Partial。store和store_tile的STPhase不能与Tile类型的scale同时使用。
原型定义
PYPTO_DECLARE_ENUM(STPhase,
Unspecified, # 不启用分阶段写回,适用于对应矩阵计算未配置AccPhase的场景
Partial, # 当前不是对该L0C Buffer计算结果的最后一次写回
Final # 当前是对该L0C Buffer计算结果的最后一次写回
)