产品支持情况
产品
|
是否支持
|
|---|
Ascend 950PR/Ascend 950DT
|
√
|
Atlas A3 训练系列产品/Atlas A3 推理系列产品
|
√
|
Atlas A2 训练系列产品/Atlas A2 推理系列产品
|
√
|
Atlas 200I/500 A2 推理产品
|
√
|
Atlas 推理系列产品AI Core
|
√
|
Atlas 推理系列产品Vector Core
|
x
|
Atlas 训练系列产品
|
√
|
功能说明
将特定TPosition的LocalTensor初始化为某一具体数值。
函数原型
| template <typename T, typename U = PrimT<T>, typename Std::enable_if<Std::is_same<PrimT<T>, U>::value, bool>::type = true>
__aicore__ inline void Fill(const LocalTensor<T>& dst, const InitConstValueParams<U>& initConstValueParams)
|
参数说明
表1 模板参数说明参数名
|
描述
|
|---|
T
|
dst的数据类型。
Atlas 训练系列产品，支持的数据类型为：half
Atlas 推理系列产品AI Core，支持的数据类型为：half/int16_t/uint16_t
Atlas A2 训练系列产品/Atlas A2 推理系列产品，支持的数据类型为：half/int16_t/uint16_t/bfloat16_t/float/int32_t/uint32_t
Atlas A3 训练系列产品/Atlas A3 推理系列产品，支持的数据类型为：half/int16_t/uint16_t/bfloat16_t/float/int32_t/uint32_t
Atlas 200I/500 A2 推理产品，支持的数据类型为：half/int16_t/uint16_t/bfloat16_t/float/int32_t/uint32_t
Ascend 950PR/Ascend 950DT，支持的数据类型为：half/int16_t/uint16_t/bfloat16_t/float/int32_t/uint32_t
|
U
|
初始化值的数据类型。
- 当dst使用基础数据类型时， U和dst的数据类型T需保持一致，否则编译失败。
- 当dst使用TensorTrait类型时，U和dst的数据类型T的LiteType需保持一致，否则编译失败。
最后一个模板参数仅用于上述数据类型检查，用户无需关注。
|
表2 参数说明参数名称
|
输入/输出
|
含义
|
|---|
dst
|
输出
|
目的操作数，结果矩阵，类型为LocalTensor。
Atlas 训练系列产品，支持的TPosition为A1/A2/B1/B2。
Atlas 推理系列产品AI Core，支持的TPosition为A1/A2/B1/B2。
Atlas A2 训练系列产品/Atlas A2 推理系列产品，支持的TPosition为A1/A2/B1/B2。
Atlas A3 训练系列产品/Atlas A3 推理系列产品，支持的TPosition为A1/A2/B1/B2。
Atlas 200I/500 A2 推理产品，支持的TPosition为A1/A2/B1/B2。
Ascend 950PR/Ascend 950DT， 支持的TPosition为A1/B1。
如果TPosition为A1/B1，起始地址需要满足32B对齐；如果TPosition为A2/B2，起始地址需要满足512B对齐。
|
InitConstValueParams
|
输入
|
初始化相关参数，类型为InitConstValueParams。
具体定义请参考${INSTALL_DIR}/include/ascendc/basic_api/interface/kernel_struct_mm.h，${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。
参数说明请参考表3。
Atlas 训练系列产品，仅支持配置迭代次数（repeatTimes）和初始化值（initValue）。
Atlas 推理系列产品AI Core，仅支持配置迭代次数（repeatTimes）和初始化值（initValue）。
Atlas A2 训练系列产品/Atlas A2 推理系列产品，支持配置所有参数。
Atlas A3 训练系列产品/Atlas A3 推理系列产品，支持配置所有参数。
Atlas 200I/500 A2 推理产品，支持配置所有参数。
Ascend 950PR/Ascend 950DT，支持配置所有参数。
- 仅支持配置迭代次数（repeatTimes）和初始化值（initValue）场景下，其他参数配置无效。每次迭代处理固定数据量（512字节），迭代间无间隔。
- 支持配置所有参数场景下，支持配置迭代次数（repeatTimes）、初始化值（initValue）、每个迭代处理的数据块个数（blockNum）和迭代间间隔（dstGap）。
|
表3 InitConstValueParams结构体参数说明参数名称
|
含义
|
|---|
repeatTimes
|
迭代次数。默认值为0。
- 仅支持配置迭代次数（repeatTimes）和初始化值（initValue）场景下，repeatTimes∈[0, 255]。
- 支持配置所有参数场景下，repeatTimes∈[0, 32767] 。
|
blockNum
|
每次迭代初始化的数据块个数，取值范围：blockNum∈[0, 32767] 。默认值为0。
- dst的位置为A1/B1时，每一个block（数据块）大小是32B；
- dst的位置为A2/B2时，每一个block（数据块）大小是512B。
|
dstGap
|
目的操作数前一个迭代结束地址到后一个迭代起始地址之间的距离。
- dst的位置为A1/B1时，单位是32B；
- dst的位置为A2/B2时，单位是512B。
取值范围：dstGap∈[0, 32767] 。默认值为0。
|
initValue
|
初始化的value值，支持的数据类型与dst保持一致。
|
调用示例
| uint32 mLength = 16;
uint32 kLength = 16;
Fill(leftMatrix, {1, static_cast<uint16_t>(mLength * kLength * sizeof(float) / 32), 0, 1}); // 给leftMatrix填充 mLength * kLength长度的数据为1， 按32B的颗粒进行填充
|