metal-kernel

作者： pytorch

為PyTorch運算元編寫Metal/MPS內核。用於為運算元添加MPS設備支援、實現Metal著色器，或將CUDA內核移植到Apple…

npx skills add https://github.com/pytorch/pytorch --skill metal-kernel

下載 ZIP GitHub

來自 pytorch 的更多技能

add-uint-support

為 PyTorch 運算子新增無號整數（uint）型別支援，透過更新 AT_DISPATCH 巨集。用於為 uint16、uint32、uint64 型別新增支援至…

調試 AOTInductor (AOTI) 錯誤與崩潰。用於遇到 AOTI 段錯誤、設備不匹配錯誤、常量加載失敗或運行時錯誤時…

將 PyTorch 的 AT_DISPATCH 巨集轉換為 ATen C++ 程式碼中的 AT_DISPATCH_V2 格式。適用於移植 AT_DISPATCH_ALL_TYPES_AND*、AT_DISPATCH_FLOATING_TYPES* 或…

distributed-triage

在 oncall:distributed 佇列中進行子分類，透過指派分散式模組標籤、轉發至子值班人員，並標記為已分類。適用於當問題具有…

為PyTorch函數和方法撰寫符合PyTorch慣例的文件字串。適用於在PyTorch程式碼中撰寫或更新文件字串時使用。

document-public-apis

透過從 docs/source/conf.py 中的 coverage_ignore_functions 和 coverage_ignore_classes 移除函式，並執行…來記錄 PyTorch 中未記錄的公開 API。

審查 PyTorch 的拉取請求，針對程式碼品質、測試覆蓋率、安全性及向後相容性。適用於審查 PR 時、被要求審查程式碼變更時…

調試 PyTorch 2 編譯器堆疊故障，包括 Dynamo 圖中斷、Inductor 程式碼生成錯誤、AOTAutograd 崩潰以及準確度不匹配。當…時使用