CANN/pypto循环结束判断API

pypto.is_loop_end

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto

产品支持情况

功能说明

判断当前迭代是否为循环的结束。

函数原型

def is_loop_end(scalar: SymInt) -> SymbolicScalar

参数说明

返回值说明

返回一个符号标量表达式，表示是否为循环结束（布尔值）

约束说明

• scalar 必须是循环迭代器返回的符号标量
• 如果不是循环索引，将抛出 ValueError 异常
• 当函数未使用 @pypto.frontend.jit 或 @pypto.frontend.function 装饰器修饰时，条件表达式需要用 pypto.cond 包装

调用示例

# 未使用装饰器，需要用 pypto.cond 包装条件表达式 def kernel(): ... for idx in pypto.loop(0, 10, 1): if pypto.cond(pypto.is_loop_end(idx)): ... # 使用装饰器，无需 pypto.cond 包装 @pypto.frontend.jit def kernel(): ... for idx in pypto.loop(0, 10, 1): if pypto.is_loop_end(idx): ...

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto