【合入来源】

如有社区issue，请关联issue链接
请勿携带内部流程信息（需求链接、问题单、内部issue等）

• 需求
• 问题单
• issue/工单
• 重构优化
• 资料更新

#2029

【修改方案】

请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列
如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容）

接入aclnnRingAttentionUpdateV2算子，算子说明文档
接入后提供接口：

【资料变更】

请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及”

涉及，接口说明文档

【接口变更】

请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及”

涉及，新增接口torch_npu.npu_ring_attention_update，接口说明文档

【功能验证】

说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤
新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图

用例覆盖

测试文件：

op-plugin/test/test_custom_ops/test_npu_ring_attention_update.py

覆盖的测试项：

Golden 逻辑：

softmax_max = max(prev_max, cur_max)
softmax_sum = prev_sum * exp(prev_max - softmax_max)
            + cur_sum * exp(cur_max - softmax_max)

attn_out = prev_attn * prev_factor + cur_attn * cur_factor

其中：

prev_factor = prev_scale * prev_sum / softmax_sum
cur_factor  = cur_scale  * cur_sum  / softmax_sum

执行结果

结果摘要：

..
----------------------------------------------------------------------
Ran 2 tests in 8.615s

OK

【CheckList】

PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x]

• 代码注释完备，正确记录错误日志
• 代码实现进行了返回值、空指针等校验
• PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等
• PR持续集成流水线（CI）执行通过，代码检查无异常