api_check_support
函数功能
查询在当前昇腾AI处理器中是否支持某接口,若配置了可选参数dtype,则表示查询当前昇腾AI处理器是否支持数据类型为dtype的某接口。
函数原型
def api_check_support(intrinsic, dtype="")
参数说明
参数名 |
类型 |
说明 |
---|---|---|
intrinsic |
string,必选 |
表示输入的接口,为字符串。
|
dtype |
string,可选 |
需要校验的接口支持的数据类型。 若为空,表示不进行数据类型的校验。 |
返回值
如果支持返回True,不支持返回False。
约束说明
使用此接口前,需要先执行set_current_compile_soc_info接口设置昇腾AI处理器的类型。若不设置,默认使用Ascend310。
调用示例
- DSL接口查询示例
- TIK接口查询示例
父主题: TBE Platform API