| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
docs: add torch.nn.attention.flex_attention.BlockMask.seq_lengths support Co-authored-by: gcw_LoYRsZyk<2394624589@qq.com> # message auto-generated for no-merge-commit merge: !43226 merge docs-blockmask-seq-lengths-support into master docs: add torch.nn.attention.flex_attention.BlockMask.seq_lengths support Created-by: gcw_LoYRsZyk Commit-by: gcw_LoYRsZyk Merged-by: ascend-robot Description: # 【合入来源】 - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [x] 资料更新 任务来源 Issue：https://gitcode.com/Ascend/pytorch/issues/3277 API 一致性验证说明：https://gitcode.com/Ascend/pytorch/issues/4943 # 【修改方案】 **API功能**： torch.nn.attention.flex_attention.BlockMask.seq_lengths 用于保存 BlockMask 对应的查询序列长度和键值序列长度，属性值为包含两个整数的 tuple。 **用例完整性**： PyTorch 社区已有 test_block_mask_vs_sequence_lengths，主要验证 BlockMask 与输入序列长度不一致时的检查逻辑，但没有直接完整覆盖 torch.nn.attention.flex_attention.BlockMask.seq_lengths 的默认推导、显式赋值、边界值及重新赋值行为。 **NPU适配**： 该属性在 PyTorch 源码中直接保存 Python tuple，torch.nn.attention.flex_attention.BlockMask.shape 通过 Python tuple 拼接使用该属性，不调用硬件算子，属于纯 Python 接口，无需新增 NPU 测试用例或进行 API 实现适配。相关说明、运行日志和截图见 Issue 4943。 **资料修改**： 补充以下版本的 API 支持资料： 1. PyTorch 2.7.1 2. PyTorch 2.9.0 3. PyTorch 2.10.0 4. PyTorch 2.11.0 5. PyTorch 2.12.0 6. PyTorch 2.13.0 7. PyTorch 2.14.0 产品支持范围为 Atlas A2训练系列产品、Atlas A3训练系列产品和 Ascend 950DT系列产品。 # 【资料变更】 在以下文件中补充 torch.nn.attention.flex_attention.BlockMask.seq_lengths 的支持情况： - docs/zh/api/native_api/pytorch_2-7-1/torch-nn-attention.md - docs/zh/api/native_api/pytorch_2-9-0/torch-nn-attention.md - docs/zh/api/native_api/pytorch_2-10-0/torch-nn-attention.md - docs/zh/api/native_api/pytorch_2-11-0/torch-nn-attention.md - docs/zh/api/native_api/pytorch_2-12-0/torch-nn-attention.md - docs/zh/api/native_api/pytorch_2-13-0/torch-nn-attention.md - docs/zh/api/native_api/pytorch_2-14-0/torch-nn-attention.md 条目位于 BlockMask 类下，放在 kv_indices 与 shape 之间，标题遵循相邻普通属性格式，不添加方法括号。torch-nn.md 中不再保留该条目。 该 API 为非计算类属性，与数据类型无关，因此不增加数据类型限制说明。 已检查七个版本的条目位置和原生文档链接，相对最新 master 仅新增七个属性区块，共 140 行，原有内容保持一致，git diff --check 通过。本 PR 仅合入 master。 # 【接口变更】 不涉及客户可见接口变更，仅补充已有 API 的支持资料。 # 【功能验证】 使用 NPU 张量构造 BlockMask，验证以下场景： 1. 不传入 seq_lengths； 2. 显式传入 seq_lengths=None； 3. 显式传入不同边界长度； 4. BLOCK_SIZE 为整数和 tuple； 5. 修改 seq_lengths 后与 BlockMask.shape 联动； 6. PyTorch 2.11.0、2.12.0的 compute_q_blocks=False； 7. BlockMask 内部非空张量位于 NPU。 核心验证代码： python import torch from torch.nn.attention.flex_attention import BlockMask from torch.testing._internal.common_utils import TestCase, run_tests device_type = ( acc.type if (acc := torch.accelerator.current_accelerator()) else "cpu" ) class TestBlockMaskSeqLengths(TestCase): def make_mask(self, **kwargs): counts = torch.full( (1, 1, 2), 3, dtype=torch.int32, device=device_type ) indices = torch.arange( 3, dtype=torch.int32, device=device_type ) indices = ( indices.reshape(1, 1, 1, 3) .expand(1, 1, 2, 3) .contiguous() ) return BlockMask.from_kv_blocks(counts, indices, **kwargs) def test_seq_lengths(self): block_mask = self.make_mask(BLOCK_SIZE=(64, 128)) self.assertEqual(block_mask.seq_lengths, (128, 384)) self.assertEqual(block_mask.shape, (1, 1, 128, 384)) self.assertEqual(block_mask.kv_indices.device.type, "npu") block_mask.seq_lengths = (128, 320) self.assertEqual(block_mask.seq_lengths, (128, 320)) self.assertEqual(block_mask.shape, (1, 1, 128, 320)) if __name__ == "__main__": run_tests() 完整验证还覆盖： python self.make_mask() self.make_mask(seq_lengths=None) self.make_mask(BLOCK_SIZE=64) self.make_mask(BLOCK_SIZE=(64, 128)) for lengths in ( (1, 1), (127, 129), (128, 128), (129, 257), (256, 384), ): block_mask = self.make_mask(seq_lengths=lengths) self.assertEqual(block_mask.seq_lengths, lengths) block_mask = self.make_mask( compute_q_blocks=False, seq_lengths=(129, 257), ) self.assertIsNone(block_mask.q_num_blocks) self.assertIsNone(block_mask.q_indices) PyTorch 2.7.1 尚无 compute_q_blocks 参数，因此该版本执行3个测试；PyTorch 2.11.0和2.12.0执行全部4个测试。 **运行日志**： | 产品 | 验证设备 | v2.7.1 | v2.11.0 | v2.12.0 | | --- | --- | --- | --- | --- | | Atlas A2 | Ascend910B4 | 3 tests, OK | 4 tests, OK | 4 tests, OK | | Atlas A3 | Ascend910_9382 | 3 tests, OK | 4 tests, OK | 4 tests, OK | | Ascend 950D | Ascend950PR_9579 | 3 tests, OK | 4 tests, OK | 4 tests, OK | **运行截图**： Atlas A2：  Atlas A3：  Ascend 950D：  PyTorch 2.13.0 当前用于刷新 native API 资料目录。运行验证基于当前已发布的 torch-npu 2.7.1、2.11.0和2.12.0完成。 # 【CheckList】 - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!43226 | 11 天前 | |
modify document Co-authored-by: m0_68000167<wangzhiyuan86@h-partners.com> # message auto-generated for no-merge-commit merge: !47886 merge master into master modify document Created-by: wzyly Commit-by: m0_68000167 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单https://gitcode.com/Ascend/op-plugin/issues/320 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】 测试环境检视问题修改+资料刷新 # 【资料变更】 涉及 # 【接口变更】 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!47886 | 13 天前 | |
docs:PyTorch Profiler文档问题修改 Co-authored-by: cai-weiwei1989<734267852@qq.com> # message auto-generated for no-merge-commit merge: !47305 merge master into master docs:PyTorch Profiler文档问题修改 Created-by: cai-weiwei1989 Commit-by: cai-weiwei1989 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） # 【资料变更】 1、格式修改 2、补全docs/zh/figures/profiler/torch_npu-profiler-schedule.png图中的skip_first_wait 3、补充numa参数和采集结果说明 # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!47305 | 28 天前 | |
support ttfhw to 30 min Co-authored-by: SCh_zx<1325467101@qq.com> # message auto-generated for no-merge-commit merge: !47890 merge master into master support ttfhw to 30 min Created-by: SCh_zx Commit-by: SCh_zx Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > 请描述修改内容的具体实现，涉及哪些组件之间进行交互，可以用1、2、3、...进行罗列\ > 如果是需求或者重构类的PR，需要补充详细设计文档（说明上下游组件关系、时序图、类图、DFX能力等内容） 进一步优化构建资料，使新用户能在30分钟内完成构建。 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!47890 | 13 天前 | |
【docs】Installation manual revised & links revised Co-authored-by: l60114168<liuyaxiong6@h-partners.com> # message auto-generated for no-merge-commit merge: !43750 merge master into master 【docs】Installation manual revised & links revised Created-by: lyx324521 Commit-by: l60114168 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ https://gitcode.com/Ascend/pytorch/issues/3327 > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】 1. 安装手册适配2.13.0 2. 修改链接 3. 调整目录结构 # 【资料变更】 涉及 # 【接口变更】 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!43750 | 2 个月前 | |
modify document Co-authored-by: m0_68000167<wangzhiyuan86@h-partners.com> # message auto-generated for no-merge-commit merge: !47886 merge master into master modify document Created-by: wzyly Commit-by: m0_68000167 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单https://gitcode.com/Ascend/op-plugin/issues/320 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】 测试环境检视问题修改+资料刷新 # 【资料变更】 涉及 # 【接口变更】 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!47886 | 13 天前 | |
modify document Co-authored-by: m0_68000167<wangzhiyuan86@h-partners.com> # message auto-generated for no-merge-commit merge: !46791 merge master into master modify document Created-by: wzyly Commit-by: m0_68000167 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [ ] 需求 - [ ] 问题单 - [x] issue/工单https://gitcode.com/Ascend/op-plugin/issues/320 - [ ] 重构优化 - [x] 资料更新 # 【修改方案】 产品名称打标签 # 【资料变更】 涉及 # 【接口变更】 不涉及 # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/pytorch!46791 | 1 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 11 天前 | ||
| 13 天前 | ||
| 28 天前 | ||
| 13 天前 | ||
| 2 个月前 | ||
| 13 天前 | ||
| 1 个月前 |