- Ascend 950PR/Ascend 950DT:支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
pypto.loop_unroll是一个支持循环展开的循环迭代器函数,功能与pypto.loop类似,增加了unroll_list参数支持多个展开方式。
[object Object]
返回一个迭代器,每次迭代产生一个元组(idx, unroll_factor),idx表示当前循环的索引值,unroll_factor标识当前选择的展开方式。
- 展开因子列表会被排序并去重,且总是包含1
- 展开因子按从大到小排序
- 每个展开因子会生成一个子循环
- 多层循环使用了loop_unroll并且配置unroll_list,会大大增加编译出的图数量,影响编译性能
[object Object]