Stage

产品支持情况

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

头文件/库文件

头文件：#include <register/register_custom_pass.h>
库文件：libregister.so

功能说明

设置自定义Pass执行阶段。

函数原型

PassRegistrationData &Stage(const CustomPassStage stage)

参数说明

参数名	输入/输出	说明
stage	输入	表示自定义Pass执行阶段。详情请参见CustomPassStage。

返回值说明

返回PassRegistrationData对象。

约束说明

无

父主题： PassRegistrationData