pypto.set_verify_golden_data

产品支持情况

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

功能说明

该接口包括以下必要功能：

• 功能1：将用户执行算子时实际的输入、输出列表设置到工具中，用于工具在粗检时使用同样的输入进行模拟计算

• 功能2：将用户已有的计算基准数据（golden）设置到工具中，用于工具在计算出模拟结果后与基准输出对比，进而粗粒度确定模拟结果的正确性

函数原型

set_verify_golden_data(in_out_tensors=None, goldens=None)

参数说明

参数名	输入/输出	说明
in_out_tensors	输入	含义：将用户执行算子时实际的输入、输出列表按照相同位置对应地设置到检测工具。 说明：jit 调用模式下，该选项不需设置 类型：List[Union(pypto.Tensor, torch.Tensor)] 取值范围：NA 默认值：NA
goldens	输入	含义：将用户已有的计算基准数据（golden）输出设置到工具中做对比检测。 说明：该列表与算子输入、输出参数列表的长度一致、位置对应。若相应位置设置为 None，表示跳过该位置的数据对比。 类型：List[Union(pypto.Tensor, torch.Tensor)] * 其中 torch.Tensor 的 device 属性需为 CPU，不支持 NPU。 取值范围：NA 默认值：NA

返回值说明

void：Set方法无返回值。设置操作成功即生效。

约束说明

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

调用示例

set_verify_golden_data(goldens=[None, None, golden_out0])
set_verify_golden_data([real_in0, real_in1, real_out0], [None, None, golden_out0])