Community编程与开发github.com

Wancha0/ascend-npu-porting

可复用的 Codex 技能,用于将 PyTorch 代码适配到华为昇腾 NPU

ascend-npu-porting 是什么?

该技能是一个专门为 Codex 设计的可复用工具,用于自动化地将基于 PyTorch 的深度学习代码迁移到华为昇腾 NPU 平台。它能够识别并替换 PyTorch 中的 CUDA 相关操作(如 `torch.cuda`、`.cuda()` 方法等),将其适配为昇腾 NPU 兼容的等价实现。此工具显著减少了手动移植的工作量,帮助开发者在昇腾硬件上快速运行 PyTorch 模型,同时保持代码逻辑的正确性和性能。支持环境包括 Codex CLI。工作原理基于静态分析,扫描代码中的 CUDA 依赖并进行系统化替换。

兼容平台~Claude Code✓Codex CLI~Cursor
npx skills add Wancha0/ascend-npu-porting

Installed? Explore more 编程与开发 skills: steipete/bluebubbles, steipete/eightctl, steipete/blucli · View all 6 →

在你喜欢的 AI 中提问

打开一个已预加载此 Agent Skill 的新对话。

文档

ascend-npu-porting 是做什么的?

可复用的 Codex 技能,用于将 PyTorch 代码适配到华为昇腾 NPU

相关技能