Wancha0/ascend-npu-porting

ascend-npu-porting 是什麼？

此技能專為開發者設計，可自動化將 PyTorch 程式碼遷移至華為昇騰 NPU 的流程。它可識別與 GPU/CUDA 相關的程式碼（如 `.cuda()`、`torch.cuda`、CUDA 設備），並將其替換為昇騰 NPU 的等效語法（例如 `.npu()`、`torch.npu`、`torch.device('npu')`）。支援多種執行平台，包括 Codex CLI、Claude Code 和 Cursor，並可透過設定 `TORCH_NPU_MIGRATION_MODE` 環境變數（設為 `migrate` 或 `non-migrate`）來控制遷移模式。此外，它還能處理資料載入器（DataLoader）的調整、修復 AMP（自動混合精度）相關問題，並更新 `requirements.txt` 以納入 `torch_npu` 相依套件。透過簡化遷移過程中常見的語法差異和環境設定問題，此技能可節省開發者的時間並減少手動錯誤。