pypto.loop

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

功能说明

定义一个循环操作，实现python当中的for循环功能。

函数原型

loop(stop: SymInt, /, **kwargs) -> Iterator[SymInt]
loop(start: SymInt, stop: SymInt, step: Optional[SymInt] = 1, /, **kwargs) -> Iterator[SymInt]

参数说明

参数名	输入/输出	说明
start	输入	循环的起始值。
stop	输入	循环的终止值。
step	输入	每次循环的步长。
**kwargs	输入	- name(str)：循环标识名称，默认生成f”loop_{loop_idx}”。 - idx_name(str): 循环索引变量的名称，默认生成f”loop_idx_{loop_idx}”。 - submit_before_loop(bool): 是否在循环开始前提交计算，默认为False。开启后会在循环开启前强制提交当前累积的计算任务到AICore执行。

返回值说明

返回一个生成器，依次生成表示每次迭代值的符号整数。

约束说明

无。

调用示例

for _ in pypto.loop(0, 10, 1, name="LOOP_L0_bIdx_mla_prolog", idx_name="b_idx"):
   ...