函数功能
查询在当前昇腾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。
|
返回值
如果支持返回True，不支持返回False。
调用示例
- 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")