Fill【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品支持Atlas A2 训练系列产品/Atlas A2 推理系列产品支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明将Global Memory上的数据初始化为指定值。该接口可用于对workspace地址或输出数据进行清零。函数原型template typename T __aicore__ inline void Fill(GlobalTensorT gmWorkspaceAddr, const uint64_t size, const T value)参数说明表1模板参数说明参数名含义T操作数的数据类型。不同型号支持的数据类型请参考支持的数据类型。表2接口参数说明参数名输入/输出含义gmWorkspaceAddr输入gmWorkspaceAddr为用户定义的全局空间是需要被初始化的空间类型为GlobalTensor。GlobalTensor数据结构的定义请参考GlobalTensor。size输入需要初始化的空间大小单位为元素个数。value输入初始化的值数据类型与gmWorkspaceAddr保持一致。返回值说明无约束说明单核调用此接口时如果后续操作涉及Unified Buffer的使用则需要在调用接口后设置MTE2流水等待MTE3流水MTE3_MTE2.md#section622mcpsimp)的同步。单核上连续多次调用此接口且初始化的值不一致时需要在接口之间设置V流水等待MTE3流水MTE3_V.md#section622mcpsimp)的同步。由于该接口内部使用同一块Unified Buffer作为中转空间进行值初始化连续调用接口时若不进行流水同步后一次写操作可能覆盖前一次未完成写入的数据导致前一次初始化Global Memory的结果非预期。当多个核调用此接口对Global Memory进行初始化时所有核对Global Memory的初始化未必会同时结束也可能存在核之间读后写、写后读以及写后写等数据依赖问题。这种使用场景下可以在本接口后调用SyncAll接口保证多核间同步正确。该接口仅支持在程序内存分配InitBuffer接口前使用。支持的数据类型Ascend 950PR/Ascend 950DT支持的数据类型为uint8_t、int8_t、uint16_t、int16_t、bfloat16_t、half、uint32_t、int32_t、float、uint64_t、int64_t。Atlas A3 训练系列产品/Atlas A3 推理系列产品支持的数据类型为uint16_t、int16_t、half、uint32_t、int32_t、float。Atlas A2 训练系列产品/Atlas A2 推理系列产品支持的数据类型为uint16_t、int16_t、half、uint32_t、int32_t、float。Atlas 推理系列产品AI Core支持的数据类型为uint16_t、int16_t、half、uint32_t、int32_t、float。调用示例本调用示例使用8个核每个核用当前blockIdx的值初始化zGm上的65536个数。// 带初始化的元素个数 constexpr int32_t INIT_SIZE 65536; // 设置源操作数在Global Memory上的起始地址为zINIT_SIZE*blockIdx所占外部存储的大小为INIT_SIZE个float zGm.SetGlobalBuffer((__gm__ float*)z INIT_SIZE * AscendC::GetBlockIdx(), INIT_SIZE); // 将起始地址为zINIT_SIZE*blockIdx和大小为INIT_SIZE个float源操作数zGm的值初始化为每个核用当前blockIdx AscendC::Fill(zGm, INIT_SIZE, (float)(AscendC::GetBlockIdx()));结果示例如下初始化后zGm数据 [0. 0. 0. ... 0. 0. 0. 1. 1. 1. ... 1. 1. 1. 2. 2. 2. ... 2. 2. 2. ... 5. 5. 5. ... 5. 5. 5. 6. 6. 6. ... 6. 6. 6. 7. 7. 7. ... 7. 7. 7.]【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考