at-dispatch-v2
作者: pytorch
將 PyTorch 的 AT_DISPATCH 巨集轉換為 ATen C++ 程式碼中的 AT_DISPATCH_V2 格式。適用於移植 AT_DISPATCH_ALL_TYPES_AND*、AT_DISPATCH_FLOATING_TYPES* 或…
npx skills add https://github.com/pytorch/pytorch --skill at-dispatch-v2來自 pytorch 的更多技能
add-uint-support
pytorch
為 PyTorch 運算子新增無號整數(uint)型別支援,透過更新 AT_DISPATCH 巨集。用於為 uint16、uint32、uint64 型別新增支援至…
official
aoti-debug
pytorch
調試 AOTInductor (AOTI) 錯誤與崩潰。用於遇到 AOTI 段錯誤、設備不匹配錯誤、常量加載失敗或運行時錯誤時…
official
distributed-triage
pytorch
在 oncall:distributed 佇列中進行子分類,透過指派分散式模組標籤、轉發至子值班人員,並標記為已分類。適用於當問題具有…
official
docstring
pytorch
為PyTorch函數和方法撰寫符合PyTorch慣例的文件字串。適用於在PyTorch程式碼中撰寫或更新文件字串時使用。
official
document-public-apis
pytorch
透過從 docs/source/conf.py 中的 coverage_ignore_functions 和 coverage_ignore_classes 移除函式,並執行…來記錄 PyTorch 中未記錄的公開 API。
official
metal-kernel
pytorch
為PyTorch運算元編寫Metal/MPS內核。用於為運算元添加MPS設備支援、實現Metal著色器,或將CUDA內核移植到Apple…
official
pr-review
pytorch
審查 PyTorch 的拉取請求,針對程式碼品質、測試覆蓋率、安全性及向後相容性。適用於審查 PR 時、被要求審查程式碼變更時…
official
pt2-bug-basher
pytorch
調試 PyTorch 2 編譯器堆疊故障,包括 Dynamo 圖中斷、Inductor 程式碼生成錯誤、AOTAutograd 崩潰以及準確度不匹配。當…時使用
official