Stage
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
头文件/库文件
- 头文件：#include <ge/fusion/pass/fusion_pass_reg.h>
- 库文件：libge_compiler.so
功能说明
融合Pass阶段定义。
注意：kAfterAssignLogicStream阶段不可注册普通融合Pass，此阶段不允许修改图结构，若误将Pass注册到此阶段，将被忽略。
函数原型
1 | FusionPassRegistrationData &Stage(CustomPassStage stage) |
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
stage |
输入 |
表示自定义融合Pass执行阶段。详情请参见CustomPassStage。 |
返回值说明
返回FusionPassRegistrationData对象。
约束说明
无