CANN/pypto循环函数API文档

pypto.loop【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto产品支持情况产品是否支持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)循环标识名称默认生成floop_{loop_idx}。- idx_name(str): 循环索引变量的名称默认生成floop_idx_{loop_idx}。- submit_before_loop(bool): 是否在循环开始前提交计算默认为False。开启后会在循环开启前强制提交当前累积的计算任务到AICore执行。返回值说明返回一个生成器依次生成表示每次迭代值的符号整数。约束说明loop返回的循环索引变量如 b_idx为SymInt类型不支持作为列表下标索引使用。调用示例for _ in pypto.loop(0, 10, 1, nameLOOP_L0_bIdx_mla_prolog, idx_nameb_idx): ...【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考