TBE基于不同的昇腾AI处理器封装了统一的硬件信息相关接口,包含获取硬件信息、设置soc信息、判断相关指令是否在当前昇腾AI处理器支持等接口,屏蔽了不同昇腾AI处理器间的差异。
接口名称
定义文件
set_current_compile_soc_info
CANN软件安装后文件存储路径中的“python/site-packages/tbe/common/platform/platform_info.py”文件。
api_check_support
get_soc_spec