pypto.pass_verify_print

产品支持情况

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

功能说明

在精度调试 Verify 特性使能时，使用该接口保存指定TTensor计算的结果到数据文件。

函数原型

pass_verify_print(*values, cond: Union[int, SymbolicScalar] = 1) -> None

参数说明

参数名	输入/输出	说明
*values	输入	含义：指定打印的数据或信息。 说明：pypto.Tensor：打印tensor的数据；int/pypto.SymbolicScalar：打印相应的值；其它Python对象：打印相应的字符串表示 类型：List[pypto.Tensor,int,pypto.SymbolicScalar,Object] 取值范围：NA 默认值：NA
cond	输入	含义：指定打印数据的满足条件 说明：表达式计算结果为1：打印指定数据；表达式计算结果为0：不打印数据 类型：Optional[int,pypto.SymbolicScalar] 取值范围：0,1 默认值：1

返回值说明

无。

约束说明

该函数需设置 pypto.set_verify_options(enable_pass_verify=True) 后生效。

调用示例

verify_options = {
        "enable_pass_verify": True,
      }

@pypto.frontend.jit(verify_options=verify_options)
def user_kernel(input0: pypto.Tensor, input1: pypto.Tensor, output: pypto.Tensor):
    ...
    for idx in pypto.loop(10):
        t0 = pypto.tensor(...)
        t1 = pypto.tensor(...)
        t2 = pypto.SOME_OP1(t0, t1)
        pypto.pass_verify_print(t2)
        t3 = pypto.SOME_OP2(t0, t2)
        pypto.pass_verify_print(t3, cond=(idx == 5))
         ...