文档
注册
评分
提单
论坛
小AI

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

返回值

如果支持返回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")
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词