提交提案之前,请先检索仓库内是否已有相同的提案,如已有请在同一提案中进行讨论。
上游 PyTorch 社区自 2.10.0 版本起,在 base API 范畴中新增了 torch.accelerator.get_device_capability 接口以及 torch.accelerator.device_index 上下文管理器类。当前 Ascend PyTorch 对应的 Native API 文档缺少上述 API 的记录与原生文档链接映射,未与 PyTorch 社区官方标准完全对齐。
torch.accelerator.get_device_capability
torch.accelerator.device_index
欢迎加入社区,感谢您对社区的贡献 🎉!
提交提案之前,请先检索仓库内是否已有相同的提案,如已有请在同一提案中进行讨论。
💻 需求背景、当前现状、期望实现的功能内容、具体的设计方案、以及测试方案
上游 PyTorch 社区自 2.10.0 版本起,在 base API 范畴中新增了
torch.accelerator.get_device_capability接口以及torch.accelerator.device_index上下文管理器类。当前 Ascend PyTorch 对应的 Native API 文档缺少上述 API 的记录与原生文档链接映射,未与 PyTorch 社区官方标准完全对齐。替代方案
补充说明
欢迎加入社区,感谢您对社区的贡献 🎉!