变更摘要

此 PR 为华为 AscendNPU SIMT 后端新增了全局 scratch 内存（global scratch memory）的分阶段（staging）支持，主要应用于 TileDotLoads 中的非加载操作数分阶段和 ConvertLayout 操作的跨布局转换。核心思路是：当共享内存（SMEM）预算不足以容纳分阶段缓冲区时，允许回退到全局内存（GMEM）路径。为此，PR 将原有的 SharedMemConflictModel 重构为更通用的 DotTilingCostModel，新增了 storeDGlobal/loadDGlobal 接口，引入了 enable-global-scratch-allocation（默认关闭）的管道选项，并重命名了 ConvertSharedPtrToMemDesc → LowerDotBuffersAndSharedMem 以更准确地反映其职责扩展。

主要改动

• 成本模型重构：SharedMemConflictModel 重命名为 DotTilingCostModel，AscendSmemGeometry 重命名为 AscendMemGeometry 并新增 peakBandwidthBytesPerCycleGmem() 方法；StagingDecisionInputs 从单一 roundTripBytes 拆分为 roundTripBytesA/roundTripBytesB，新增 smemStageA/smemStageB 标志和 stagingFixedOverheadCyclesGmem 字段，stageThroughSmem 改为 stageThroughMem，使成本模型能分别评估 SMEM 和 GMEM 路径的开销。

• 新增全局内存加载/存储接口：在 TargetInfo 和 TargetInfoBase 中新增 storeDGlobal 和 loadDGlobal 虚方法，TargetInfo.cpp 中实现了完整的全局内存存取逻辑，支持向量化、子字节元素处理、大宽度拆分，并使用 ascend_dpx::StoreOp/ascend_dpx::LoadOp 及 L2_CACHE_HINT_NORMAL_FV 缓存策略。

• StageNonLoadOperandPattern 支持全局 scratch：当 totalStageBytes > smemBudgetBytes 且 allowGlobalScratch 为真时，模式不再直接失败，而是按需将 A 或 B 操作数的地址空间切换为 kGlobalMemoryAddressSpace（ptr<1>），并通过 appendScratchArg 向函数追加带 bishengir.scratch_global 标记的参数，使非加载操作数可通过全局内存完成分阶段。

• ConvertLayoutOp 全局内存路径：新增 ConvertLayoutOpConversionGM 模式，在 Allocation.cpp 中当 allowGlobalScratch 开启且 ConvertLayoutOp 的共享内存偏移量超出 smemSize 时，将其标记为 store_to_gmem 并分配全局 scratch 偏移；lowerLdStShared 新增 switchToGM 参数，在 GMEM 路径下调用 storeDGlobal/loadDGlobal 替代 storeDShared/loadDShared。

• 管道与 Pass 重命名：ConvertSharedPtrToMemDesc pass 重命名为 LowerDotBuffersAndSharedMem（lower-dot-buffers-and-shared-mem），新增 SetAllowGlobalScratchAttr pass（set-allow-global-scratch-attr）用于将命令行选项写入模块属性 ttg.enable-global-scratch-allocation；管道中条件性插入 TritonGPUGlobalScratchAllocationPass。

libing-pipeline创建中，请等待……

代码审查

✅ 未发现问题

Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here

PR Approval Progress

✅ Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

💡 Tip:

• Committer can comment /approve or /lgtm
• Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

SanidhyaBarara, thanks for your pull request. All authors of the commits have signed the CLA. 👍

ascend docs pipeline is running...

openLiBing协同任务执行成功, 任务链接-https://www.openlibing.com/apps/transitionPrDetail?owner=Ascend&repo=ascendnpu-ir&number=2333&platform=gitcode&target=codeCooperate&codeHostingPlatformFlag=gitcode

✅ 文档门禁通过！

compile

ascend docs pipeline is running...

✅ 文档门禁通过！

npuir-smoke

/lgtm

/approve

Not enough required testers

If you want to solve this problem, you can click here to do it in the FAQs.