CANN opbase：GetSimtMaxDCacheSize 获取平台 SIMT 最大 DCache 大小接口详解

CANN opbaseGetSimtMaxDCacheSize 获取平台 SIMT 最大 DCache 大小接口详解【免费下载链接】opbase本项目是CANN算子库的基础框架库为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbaseGetSimtMaxDCacheSize是 CANN opbase 基础框架库op_common平台信息工具组中的一个 Tiling 侧接口用于在 Host 端 Tiling 编译过程中获取平台 SIMT 最大 DCache 大小单位 bytes。本文基于该接口的 API 文档结合仓库中的头文件实现与接口索引完整讲解其函数原型、参数与返回值语义、源码级实现细节以及它在platform工具组中的定位与调用方式帮助算子开发者在 Tiling 代码中正确查询并消费这一平台属性。接口定位op_common 的 platform 工具组在 opbase 仓库中GetSimtMaxDCacheSize归属于op_common公共依赖库的platform功能组。根据 op_common API 介绍文档platform 组统一实现于op_common/op_host/util/platform_util.h与GetSimtMaxDCacheSize同组、同头文件的接口包括GetAivCoreNum获取平台 AI Vector 的核数GetAicCoreNum获取平台 AI Cube 的核数GetUbSize/GetUbBlockSize获取 UBUnified Buffer空间大小与 block 单元大小GetVRegSize获取平台向量寄存器Vector Register的大小GetSimtMaxThreadNum获取平台 SIMT 最大线程数GetSimtMaxDCacheSize获取平台 SIMT 最大 DCache 大小GetCacheLineSize/GetSectorCacheLineSize获取 Cache Line 与 Sector Cache Line 大小GetNddmaDcacheSize获取 NDDMA DCache 的大小GetWorkspaceSize获取平台 LibApi Workspace 的大小。该组的完整导航见 platform 接口列表接口汇总入口见 op_common 接口清单。整个 platform 组的设计目的一致为 Host 端 Tiling 逻辑提供统一、与具体平台解耦的平台硬件属性查询入口避免算子 Tiling 代码中散落硬编码的硬件常量。函数原型与参数说明官方 API 文档定义的函数原型如下template typename T uint32_t GetSimtMaxDCacheSize([[maybe_unused]] const T *context)参数名输入/输出说明context输入Tiling 的上下文信息类型为TilingContext/TilingParseContext。模板参数 T接口采用模板形式使其同时兼容TilingContext与TilingParseContext两类上下文类型开发者无需为两种场景各写一份查询代码。[[maybe_unused]]标注当前实现中该参数实际未被读取返回值是固定常量因此形参被显式标注为可能未使用以消除编译告警。这一标注也暗示了接口的演进路径——参数签名保留不变便于未来真正接入平台信息查询时无需破坏调用方。返回值uint32_t返回平台 SIMT 最大 DCache 大小当前值为 128 * 1024 B 131072 bytes。约束说明无。源码实现常量返回的只读查询在仓库中该接口的实现位于 platform_util.h头文件内联实现无独立.cpp文件/** * Get the maximum Dcache size used by simt in bytes: 128 * 1024B */ template typename T uint32_t GetSimtMaxDCacheSize([[maybe_unused]] const T* context) { return 131072U; }从源码结构看实现上有几个值得注意的点纯常量返回。函数体直接返回字面量131072U与 API 文档中“当前为 128 * 1024B 131072 bytes”的返回值说明完全一致。这意味着当前版本下该值不随芯片型号或上下文变化属于“当前平台代际的固定规格”。与组内动态查询接口形成对比。同文件中如GetAivCoreNum、GetUbSize、GetWorkspaceSize等接口会通过context-GetPlatformInfo()获取平台信息再用platform_ascendc::PlatformAscendC动态查询并附带static_assert约束 T 只能是TilingParseContext或TilingContext以及空指针检查。而GetSimtMaxDCacheSize、GetSimtMaxThreadNum固定返回 2048、GetCacheLineSize固定返回 256、GetNddmaDcacheSize固定返回 8192等接口见 platform_util.h则采用固定常量实现不访问 context。可以推断当某个硬件属性在当前支持的平台范围内取值一致、或底层查询 API 尚未就绪时接口先以常量落地同时保留统一的context签名为后续切换为动态查询预留空间同文件的GetUbBlockSize注释 “will using AscendC api later” 即为同类演进示例。头文件内联、零运行时依赖。实现直接在 public 头文件中以内联模板函数形式提供算子 Tiling 库只需包含该头文件即可使用无需链接额外的实现库。调用示例与典型使用场景API 文档给出的关键调用示例如下仅供参考不支持直接拷贝运行compileInfo-simtMaxDCacheSize GetSimtMaxDCacheSize(context_);从示例代码结构看其典型用法是在 Tiling 编译/解析阶段将查询结果写入编译信息结构体示例中的compileInfo的成员simtMaxDCacheSize供后续编译或运行逻辑消费。配套的姊妹接口 GetSimtMaxThreadNum 提供了完全对称的用法示例compileInfo-simtMaxThreadNum GetSimtMaxThreadNum(context_);两者组合使用可获得描述 SIMT 执行资源的两项关键属性属性查询接口当前值含义SIMT 最大线程数GetSimtMaxThreadNum2048单块可调度线程上限用于线程块切分决策SIMT 最大 DCache 大小GetSimtMaxDCacheSize131072 B128 KBSIMT 可使用的最大数据缓存容量用于缓存策略与数据分块决策在 Tiling 逻辑中这两个值通常用于估算线程块数量、确定每次搬运/缓存的数据量上限从而把算子的数据切分尺寸约束在平台缓存能力之内减少重复访问片外内存。开发者应将GetSimtMaxDCacheSize(context)的返回值视为“可安全利用的最大 DCache 容量上限”切分或缓存规划时若需要精确的可用额度应在此上限基础上结合自身实现的占用开销自行预留。与周边接口的配合与注意事项参数类型context传入TilingContext或TilingParseContext均可由于当前实现不读取该参数传参主要起类型占位与未来兼容作用。取值稳定性文档与源码一致表明当前返回 131072 bytes若未来平台规格变化或接口切换为动态查询该值可能随之变化Tiling 代码不应在别处重复硬编码 131072而应统一通过本接口获取。无约束、无错误码路径接口无前置约束不返回错误状态属于“取即所得”的只读查询这与同组中需要空指针检查的动态接口如GetWorkspaceSize不同。相关接口交叉参考涉及缓存与搬运的其他平台属性可配合 GetNddmaDcacheSizeNDDMA DCache 大小当前 8192 B、GetCacheLineSize 与 GetSectorCacheLineSizeCache Line 为 256 BSector Cache Line 为其一半共同使用构成完整的平台缓存属性视图。小结GetSimtMaxDCacheSize是 opbaseop_commonplatform 工具组中面向 Tiling 场景的 SIMT 缓存容量查询接口原型为模板函数接受TilingContext/TilingParseContext上下文当前实现固定返回 128 KB131072 bytes。其实现位于 platform_util.h以头文件内联形式随 public 头文件交付调用方通过compileInfo-simtMaxDCacheSize GetSimtMaxDCacheSize(context_)的典型模式将其接入 Tiling 编译信息与GetSimtMaxThreadNum等姊妹接口共同支撑算子 Tiling 的平台感知切分策略。【免费下载链接】opbase本项目是CANN算子库的基础框架库为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考