评分并提供意见反馈 :
华为采用机器翻译与人工审校相结合的方式将此文档翻译成不同语言,希望能帮助您更容易理解此文档的内容。 请注意:即使是最好的机器翻译,其准确度也不及专业翻译人员的水平。 华为对于翻译的准确性不承担任何责任,并建议您参考英文文档(已提供链接)。
api_check_support
函数功能
查询在当前昇腾AI处理器中是否支持某接口,若配置了可选参数dtype,则表示查询当前昇腾AI处理器是否支持数据类型为dtype的某接口。
函数原型
def api_check_support(intrinsic, dtype="")
参数说明
参数名 |
类型 |
说明 |
---|---|---|
intrinsic |
string,必选 |
表示输入的接口,为字符串。
|
dtype |
string,可选 |
需要校验的接口支持的数据类型。 若为空,表示不进行数据类型的校验。
|
返回值
参数名 |
类型 |
说明 |
---|---|---|
whether_support |
bool |
如果支持返回True,不支持返回False。 |
约束说明
无
调用示例
- DSL接口查询示例
te.platform.api_check_support("te.lang.cce.vmax", dtype="float16")
具体支持的接口及数据类型,详见TBE DSL API。
- TIK接口查询示例
te.platform.api_check_support("tik.vec_add", dtype="float16")
具体支持的接口及数据类型,详见TIK API。