【合入来源】

• 需求
• 问题单 https://gitcode.com/Ascend/pytorch/issues/2253
• issue/工单
• 重构优化
• 资料更新

【修改方案】

修复部分测试用例失败问题，当前提交包含两类调整：

1. test_npu_multi_head_attention.py

   • 去掉 test_npu_multi_head_attention 上的 skip 标记，恢复用例执行。
   • 在用例开始处设置：

     torch.npu.config.allow_internal_format = True
     

   • 根因是该用例构造了 FORMAT_NZ 输入，npu_multi_head_attention 相关路径依赖内部 format；未开启 internal format 时，CPU 参考链路与 NPU 算子链路的 format 行为不一致，导致 assertRtolEqual 精度对比失败。
   • 删除不再使用的 import unittest。

2. .pytorch-disabled-tests.json

   • 从 disabled 列表中移除 _chunk_cat 相关用例：
     • test_compare_cpu__chunk_cat_npu_float32
     • test_variant_consistency_eager__chunk_cat_npu_float32
     • test_operator__chunk_cat_npu_float32
   • _chunk_cat 相关问题已修复，恢复这些用例进入常规测试看护。

【资料变更】

不涉及。

【接口变更】

不涉及对外接口变更。

本次仅调整测试用例配置和测试执行条件，不新增公开 API，不修改已有接口语义。

【功能验证】

• test/custom_ops/test_npu_multi_head_attention.py
  
• test_compare_cpu__chunk_cat_npu_float32
  
• test_variant_consistency_eager__chunk_cat_npu_float32
  
• test_operator__chunk_cat_npu_float32
  

【CheckList】

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