CANN/asc-devkit平台信息添加API
AddPlatformInfo【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明设置硬件平台信息便于用户在算子Tiling函数调测中使用。支持以下两种设置方式自动获取当前硬件平台信息传入空指针自动获取当前硬件信息并添加到ContextBuilder类中。指定硬件平台信息传入具体的AI处理器型号添加对应硬件信息至ContextBuilder类中。若设置失败会打印报错信息。关于日志配置和查看请参考。函数原型ContextBuilder AddPlatformInfo(const char* customSocVersion)参数说明参数名输入/输出描述customSocVersion输入AI处理器型号。配置方式如下针对如下产品在安装AI处理器的服务器执行npu-smi info命令进行查询获取Name信息。实际配置值为AscendName例如Name取值为xxxyy实际配置值为Ascendxxxyy。Atlas A2 训练系列产品 / Atlas A2 推理系列产品针对如下产品在安装AI处理器的服务器执行npu-smi info -t board -iid-cchip_id命令进行查询获取Chip Name和NPU Name信息实际配置值为Chip Name_NPU Name。例如Chip Name取值为AscendxxxNPU Name取值为1234实际配置值为Ascendxxx_1234。其中id设备id通过npu-smi info -l命令查出的NPU ID即为设备id。chip_id芯片id通过npu-smi info -m命令查出的Chip ID即为芯片id。Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品 / Atlas A3 推理系列产品返回值说明当前ContextBuilder对象。约束说明AddPlatformInfo调用后需要通过BuildTilingContext来构建Tiling的上下文并传递给Tiling函数来使用。调用示例void AddPlatformInfoDemo(......) { auto holder context_ascendc::ContextBuilder() // ... ... // 增加算子输入输出接口的调用 .AddPlatformInfo(Ascendxxxyy) .BuildTilingContext(); auto tilingContext holder-GetContextgert::TilingContext(); // ... ... }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考