AGENT SKILLS · MISSIONS-AUSRÜSTUNG / pytorch

pytorch · Agent Skills

8 Skills mit ausführlicher Dokumentation und geprüfter Quelle. Öffne eine Karte, um die Originalanleitung, den Paketinhalt und die Installation zu prüfen.

pytorch
Offiziell

add-uint-support

Add unsigned integer (uint) type support to PyTorch operators by updating ATDISPATCH macros. Use when adding support for uint16, uint32, uint64 types to operators, kernels, or when user mentions enabling unsigned types, barebones unsigned types, or uint support.

Installationen / 1h
2
GitHub Stars
104.945
Aktualisiert
20. Okt.
pytorch
Offiziell

add-op-overload

Checklist for adding a new overload (e.g. mm.dtype, addmm.computemode) to an existing ATen op in nativefunctions.yaml. Use before the first rebuild when adding an overload or new kwarg variant of an existing operator, so every rebuild-triggering edit lands in one build and packet-level Python registrations (meta, decompositions, Inductor lowerings, DTensor) don't silently capture the new overload.

Installationen / 1h
1
GitHub Stars
105.080
Aktualisiert
26. Okt.
pytorch
Offiziell

at-dispatch-v2

Convert PyTorch ATDISPATCH macros to ATDISPATCHV2 format in ATen C++ code. Use when porting ATDISPATCHALLTYPESAND, ATDISPATCHFLOATINGTYPES, or other dispatch macros to the new v2 API. For ATen kernel files, CUDA kernels, and native operator implementations.

Installationen / 1h
1
GitHub Stars
105.080
Aktualisiert
26. Okt.
pytorch
Offiziell

docstring

Write docstrings for PyTorch functions and methods following PyTorch conventions. Use when writing or updating docstrings in PyTorch code.

Installationen / 1h
1
GitHub Stars
103.248
Aktualisiert
24. Sept.
pytorch
Offiziell

metal-kernel

Write Metal/MPS kernels for PyTorch operators. Use when adding MPS device support to operators, implementing Metal shaders, or porting CUDA kernels to Apple Silicon. Covers nativefunctions.yaml dispatch, host-side operators, and Metal kernel implementation.

Installationen / 1h
1
GitHub Stars
103.248
Aktualisiert
24. Sept.
pytorch
Offiziell

pyrefly-type-coverage

Migrate a file to use stricter Pyrefly type checking with annotations required for all functions, classes, and attributes.

Installationen / 1h
1
GitHub Stars
105.080
Aktualisiert
26. Okt.
pytorch
Offiziell

scrub-issue

Fetch, analyze, reproduce, and minimize GitHub issue reproductions. Use only when the user asks to check whether a GitHub issue reproduces, minimize a GitHub issue's repro, analyze a GitHub bug report, or scrub/triage a GitHub issue for reproducibility. Do not use for standalone reproduction requests without a GitHub issue.

Installationen / 1h
1
GitHub Stars
105.080
Aktualisiert
26. Okt.
pytorch
Offiziell

triaging-issues

Triages GitHub issues by routing to oncall teams, applying labels, and closing questions. Use when processing new PyTorch issues or when asked to triage an issue.

Installationen / 1h
1
GitHub Stars
103.248
Aktualisiert
24. Sept.