该问题是怎么引起的？

DSA 相关分布式算子（lightning_indexer、npu_dense_lightning_indexer_softmax_lse、npu_dense_lightning_indexer_grad_kl_loss、npu_sparse_lightning_indexer_grad_kl_loss、npu_sparse_flash_attention）在 preprocess 中对 actual_seq_len（q/kv 的累积序列长度）入参无条件调用 .to_local()，即要求它必须是 Replicate 的 DTensor。

但 actual_seq_len 是网络在前向过程中构造出来的普通 Tensor，用户调用分布式算子时无法保证把它包装成 DTensor。当传入普通 Tensor 时，.to_local() 抛 AttributeError。

现状（根因分析）

各算子 preprocess 对 actual_seq_len 的处理存在两类问题：

两个矛盾点：

1. softmax_lse / lightning_indexer 对 actual_seq_len 直接 .to_local()，不兼容网络构造的普通 Tensor。
2. 另外三个算子把宽松 _to_local（isinstance(DTensor) 判断）套在所有入参上，导致 actual_seq_len 之外的主张量（query/key/value/index/weights、rope 等）即使传普通 Tensor 也被静默透传，而这些入参必须是 DTensor 才能推导 layout。

修复方案

1. 新增宽松 helper _to_local_seq_len：仅当入参是 DTensor 时 to_local，普通 Tensor / None 原样透传。只用于 actual_seq_len。
2. 其余参与 layout 推导的入参收紧为严格 _to_local：None 透传（可选参数未提供），否则直接 to_local，普通 Tensor 触发 AttributeError（明确拒绝，避免静默放行）。
3. 同步更新 UT/ST：ST 把 actual_seq_len 从 Replicate DTensor 改为普通 Tensor 传入；UT 增加「普通 Tensor 透传」「主张量拒绝普通 Tensor」的回归用例。

重现步骤

1. 构造 query / key / weights 为 Replicate 或 Shard 的 DTensor。
2. actual_seq_qlen / actual_seq_klen 用网络构造的普通 Tensor（非 DTensor）。
3. 调用 npu_dense_lightning_indexer_softmax_lse(q, k, w, actual_seq_qlen=..., actual_seq_klen=..., layout='TND')。
4. preprocess 中 actual_seq_qlen.to_local() 触发报错。

报错信息

AttributeError: 'Tensor' object has no attribute 'to_local'

关联 PR

mindspore/hyper-parallel#871