描述

为支持从统一缓冲区（Unified Buffer, UB）到全局内存（Global Memory, GM）的高效数据散射（Index Put）操作，实现按指定索引将数据精准写入目标张量，新增 libdevice.index_put 接口。TA 层在代码生成阶段会构造 func.call "triton_index_put" 调用，底层对接 SIMT 架构的模板实现，适配硬件级别的高效数据搬移需求，满足 2~5 维张量的索引写入场景。

接口说明

index_put 接口用于将 UB 中的 value 张量，按照 UB 中的 index 索引，沿目标张量的指定维度 dim，写入到 GM 中的 ptr 目标张量。

参数说明

约束与限制

• 维度一致性：ptr（目标张量）与 value（待写入张量）的秩（rank）必须相同。
• 数据类型限制：ptr 和 value 的数据类型仅支持 float16、bfloat16、float32 三种浮点类型。
• 索引张量约束：index 必须为整数类型张量，且index 固定为 1D 维度。
• 维度范围约束：value 仅支持 2~5 维张量；dim 需为有效维度（满足 0 ≤ dim < rank(value)-1），确保沿合法维度进行散射。

checklist

• 是否通过本地IDE对代码进行静态检查
• 是否通过本地IDE对代码进行格式化处理
• 是否进行空指针校验
• 是否进行返回值校验
• 是否正确释放new/malloc申请的内存
• 是否充分考虑接口的异常场景
• 是否正确记录错误日志