api_check_support

函数功能

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

函数原型

def api_check_support(intrinsic, dtype="")

参数说明

返回值

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

约束说明

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

调用示例

• DSL接口查询示例
  本示例中，以昇腾AI处理器的型号为“Ascend910”为例，实际调用时，请将变量soc_version的值修改为实际的昇腾AI处理器型号。

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

• TIK接口查询示例
  本示例中，以昇腾AI处理器的型号为“Ascend910”为例，实际调用时，请将变量soc_version的值修改为实际的昇腾AI处理器型号。

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