api_check_support

函数功能

查询在当前昇腾AI处理器中是否支持某接口，若配置了可选参数dtype，则表示查询当前昇腾AI处理器是否支持数据类型为dtype的某接口。

函数原型

def api_check_support(intrinsic, dtype="")

参数说明

参数名	类型	说明
intrinsic	string，必选	表示输入的接口，为字符串。 DSL接口，例如：tbe.dsl.vmax TIK接口，例如：tik.vec_add
dtype	string，可选	需要校验的接口支持的数据类型。若为空，表示不进行数据类型的校验。数据类型的范围： float16, float32, int8, int16, int32, int64, uint8, uint16, uint32, uint64。若API接口中输入/输出的数据类型不同，则按照输出数据类型、输入数据1类型、输入数据2类型……的简写进行组合。例如，输出数据类型为float32，两个输入数据的类型为float16，则dtype填为：f32f16f16。对于TIK API中的vconv这类进行数据类型转换的指令，按照：“<输入数据类型简写>2<输出数据类型简写><转换模式简写（可选）>”进行组合。例如：输入数据类型为float16，输出数据类型为float32，则dtype配置为f162f32。输入数据类型为float16，输出数据类型为int32，转换模式为“floor”，则dtype配置为f162s32f。

参数名

类型

说明

intrinsic

string，必选

表示输入的接口，为字符串。

DSL接口，例如：tbe.dsl.vmax
TIK接口，例如：tik.vec_add

dtype

string，可选

需要校验的接口支持的数据类型。

若为空，表示不进行数据类型的校验。

数据类型的范围：
float16, float32, int8, int16, int32, int64, uint8, uint16, uint32, uint64。
若API接口中输入/输出的数据类型不同，则按照输出数据类型、输入数据1类型、输入数据2类型……的简写进行组合。
例如，输出数据类型为float32，两个输入数据的类型为float16，则dtype填为：f32f16f16。
对于TIK API中的vconv这类进行数据类型转换的指令，按照：“<输入数据类型简写>2<输出数据类型简写><转换模式简写（可选）>”进行组合。
例如：
- 输入数据类型为float16，输出数据类型为float32，则dtype配置为f162f32。
- 输入数据类型为float16，输出数据类型为int32，转换模式为“floor”，则dtype配置为f162s32f。

返回值

如果支持返回True，不支持返回False。

约束说明

使用此接口前，需要先执行set_current_compile_soc_info接口设置昇腾AI处理器的类型。若不设置，默认使用Ascend310。

调用示例

DSL接口查询示例

实际调用时，请将变量soc_version的值修改为实际的昇腾AI处理器型号。

import tbe
soc_version="xxx"
tbe.common.platform.set_current_compile_soc_info(soc_version)
tbe.common.platform.api_check_support("tbe.dsl.vmax", dtype="float16")

TIK接口查询示例

实际调用时，请将变量soc_version的值修改为实际的昇腾AI处理器型号。

import tbe
soc_version="xxx"
tbe.common.platform.set_current_compile_soc_info(soc_version)
tbe.common.platform.api_check_support("tik.vec_add", dtype="float16")

父主题： TBE Platform API