DataCopyMode

产品支持情况

• Ascend 950PR/Ascend 950DT：支持

• Atlas A3 训练系列产品/Atlas A3 推理系列产品：不支持

• Atlas A2 训练系列产品/Atlas A2 推理系列产品：不支持

功能说明

DataCopyMode定义了vf.load_align、vf.store_align和vf.gather的数据拷贝粒度模式，用于控制是逐元素拷贝还是按DataBlock块拷贝。

原型定义

class DataCopyMode(enum.Enum):
     NORM = ...  # 普通逐元素拷贝（默认）
     DATA_BLOCK_LOAD = ...  # DataBlock加载（用于vf.gather：按32B DataBlock粒度gather）
     DATA_BLOCK_COPY = ...  # 非连续DataBlock拷贝（基于block_stride）

调用示例

import pypto_pro.language as pl

@pl.vector_function
def vf_kernel():
    vf.store_align(ub_tile, reg, data_copy_mode=pl.DataCopyMode.DATA_BLOCK_COPY)