asc.language.basic.init_const_value【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyascasc.language.basic.init_const_value(dst: LocalTensor, init_const_value_params: InitConstValueParams) → None将特定TPosition的LocalTensor初始化为某一具体数值。对应的Ascend C函数原型template typename T, typename U PrimTT, typename Std::enable_ifStd::is_samePrimTT, U::value, bool::type true __aicore__ inline void InitConstValue(const LocalTensorT dst, const InitConstValueParamsU initConstValueParams)参数说明dst目的操作数类型为 LocalTensor支持的 TPosition 为 A1/A2/B1/B2。如果TPosition为A1/B1起始地址需要满足32B对齐如果TPosition为A2/B2起始地址需要满足512B对齐。init_const_value_params初始化相关参数类型为InitConstValueParams。仅支持配置迭代次数repeat_times和初始化值init_value场景下其他参数配置无效。每次迭代处理固定数据量512字节迭代间无间隔。支持配置所有参数场景下支持配置迭代次数repeat_times、初始化值init_value、每个迭代处理的数据块个数block_num和迭代间间隔dst_gap。repeat_times迭代次数。默认值为0。仅支持配置迭代次数repeat_times和初始化值init_value场景下repeat_times∈[0, 255]。支持配置所有参数场景下repeat_times∈[0, 32767]。block_num每次迭代初始化的数据块个数取值范围block_num∈[0, 32767] 。默认值为0。dst的位置为A1/B1时每一个block数据块大小是32Bdst的位置为A2/B2时每一个block数据块大小是512B。dst_gap目的操作数前一个迭代结束地址到后一个迭代起始地址之间的距离。取值范围dst_gap∈[0, 32767] 。默认值为0。dst的位置为A1/B1时单位是32Bdst的位置为A2/B2时单位是512B。init_value初始化的value值支持的数据类型与dst保持一致。约束说明操作数地址对齐要求请参见 《Ascend C算子开发接口》 中的“通用说明和约束-通用地址对齐约束”。调用示例import asc dst asc.LocalTensor(dtypeasc.float32, posasc.TPosition.A1, addr0, tile_size128) params asc.InitConstValueParams(repeat_times1, block_num2, dst_gap0, init_value2.2) asc.init_const_value(dstdst, init_const_value_paramsparams)【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyasc创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考