接口列表
主要介绍Scope融合规则开发接口。
您可以在“Ascend-cann-toolkit安装目录/ascend-toolkit/latest/compiler/include/register/scope/scope_fusion_pass_register.h”查看接口定义。
如下接口对应的库文件是:libregister.so。
Scope类
接口名称 |
简介 |
---|---|
Scope构造函数和析构函数。 |
|
初始化Scope对象。 |
|
获取Scope的名称。 |
|
获取Scope子类型名称。 |
|
获取和Scope相关联的nodes map。 |
|
获取Scope的子Scope信息。 |
|
获取Scope名称字符串中,最后一个“/”分隔符后的字符串。 |
|
获取Scope关联的子Scope,子Scope关联的子Scope,层层递归下去,取得所有相关联的Scope集合。 |
|
获取Scope的父Scope信息。 |
FusionScopesResult类
接口名称 |
简介 |
---|---|
FusionScopesResult构造函数和析构函数。 |
|
初始化FusionScopesResult对象。 |
|
设置Scope result的名称。 |
|
设置Scope result的类型。 |
|
设置Scope result的描述信息。 |
|
获取Scope result的名称。 |
|
获取Scope result中的nodes信息。 |
|
在FusionScopesResult中插入融合算子的输入信息。 |
|
在FusionScopesResult中插入融合算子的输出信息。 |
|
设置InnerNodeInfo的名称。 |
|
设置InnerNodeInfo的类型。 |
|
在InnerNodeInfo中插入输入算子相关信息。 |
|
在InnerNodeInfo中插入输出算子相关信息。 |
|
构建InnerNode。 |
|
设置InnerNode输入格式。 |
|
设置InnerNode输出算子格式。 |
|
设置DynamicInput格式。 |
|
设置DynamicOutput格式。 |
|
定义Operator信息。 |
|
获取InnerNodeInfo名。 |
|
获取InnerNodeInfo类型。 |
|
获取InnerNodeInfo输入信息。 |
|
获取InnerNodeInfo输出信息。 |
|
InnerNodeInfo构造函数和析构函数。 |
|
特殊定义的类型,融合结果中配置了这个类型表示希望将scope映射成多个D算子。 |
|
特殊定义的类型,融合结果中配置了这个类型表示有错误发生。 |
|
表示小算子的输入来自scope边界。 |
|
表示小算子输入到scope边界。 |
|
检查FusionScopesResult是否满足条件。 |
|
增加InnerNode节点。 |
|
最近添加的InnerNode节点。 |
|
获取指定index的InnerNode,获取之后可以对该InnerNode进行修改。 |
|
检查InnerNodeInfo输入输出。 |
ScopeTree类
接口名称 |
简介 |
---|---|
ScopeTree构造函数和析构函数。 |
|
初始化ScopeTree对象。 |
|
获取ScopeTree中包含的所有Scope信息。 |
ScopeGraph类
接口名称 |
简介 |
---|---|
ScopeGraph构造函数和析构函数。 |
|
初始化ScopeGraph对象。 |
|
获取ScopeTree对象实例。 |
|
获取ScopeGraph中包含的nodes信息。 |
ScopeAttrValue类
接口名称 |
简介 |
---|---|
ScopeAttrValue构造函数和析构函数。 |
|
设置ScopeAttrValue的整型值。 |
|
设置ScopeAttrValue的浮点类型值。 |
|
设置ScopeAttrValue的string类型值。 |
|
设置ScopeAttrValue的布尔类型值。 |
ScopeBaseFeature类
接口名称 |
简介 |
---|---|
ScopeBaseFeature析构函数。 |
|
Scope融合匹配接口。 |
NodeOpTypeFeature类
接口名称 |
简介 |
---|---|
NodeOpTypeFeature构造函数和析构函数。 |
|
Scope融合匹配接口。 |
NodeAttrFeature类
接口名称 |
简介 |
---|---|
NodeAttrFeature构造函数和析构函数。 |
|
Scope融合匹配接口。 |
ScopeFeature类
接口名称 |
简介 |
---|---|
ScopeFeature构造函数和析构函数。 |
|
Scope融合匹配接口。 |
ScopePattern类
接口名称 |
简介 |
---|---|
ScopePattern构造函数和析构函数。 |
|
Scope匹配到融合规则之后,设置该Scope的类型。 |
|
添加NodeOpTypeFeature对象到ScopePattern。 |
|
添加NodeAttrFeature对象到ScopePattern。 |
|
添加ScopeFeature对象到ScopePattern。 |
ScopesResult类
接口名称 |
简介 |
---|---|
ScopesResult构造函数和析构函数。 |
|
设置ScopeResult中包含的scopes。 |
|
设置ScopeResult中包含的nodes。 |
ScopeBasePass类
接口名称 |
简介 |
---|---|
ScopeBasePass构造函数和析构函数。 |
|
方法定义,子类实现各自的融合策略并构建Patterns。 |
|
方法定义,定义scope融合规则的名称。 |
|
方法定义,子类实现scope匹配及运算符融合方法。 |
|
方法定义,子类实现各自的融合结果,并设置最终融合算子的输入和输出。 |
ScopeUtil类
接口名称 |
简介 |
---|---|
将字符串str中包含old_value的字符串用new_value替换。 |
|
释放ScopePattern集的资源。 |
|
释放一个batch包含的ScopePattern资源。 |
ScopeFusionPassRegistry类
接口名称 |
简介 |
---|---|
ScopeFusionPassRegistry析构函数。 |
|
获得ScopeFusionPassRegistry单例对象。 |
|
注册融合算子。 |
ScopeFusionPassRegistrar类
接口名称 |
简介 |
---|---|
ScopeFusionPassRegistrar构造函数和析构函数。 |
宏
接口名称 |
简介 |
---|---|
注册Scope融合算子的宏。 |