Skip to content

Pull requests: pytorch/ao

New pull request New
117 Open 1,319 Closed
117 Open 1,319 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

enforce AOBaseConfig type in quantize_'s config argument CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: bc-breaking Use this tag if this PR breaks backward compatibility
#1861 opened Mar 10, 2025 by vkuzo Loading…
2
migrate prototype codebook quant to configs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
#1858 opened Mar 7, 2025 by vkuzo Loading…
2
migrate nf4 to configs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
#1857 opened Mar 7, 2025 by vkuzo Loading…
2
migrate sparsify_ to configs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#1856 opened Mar 7, 2025 by vkuzo Loading…
2
migrate prototype/quantized_training to configs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
#1855 opened Mar 7, 2025 by vkuzo Loading…
5
Add stack for qkv projection CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1848 opened Mar 7, 2025 by drisspg • Draft
2
[wip] sparse marlin rocm compilation ciflow/rocm CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: rocm
#1847 opened Mar 6, 2025 by jcaip • Draft
3
Migrate to int args CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1846 opened Mar 5, 2025 by metascroy Loading…
1
Add bias support for Int8DynActInt4WeightLinear CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#1845 opened Mar 5, 2025 by andrewor14 Loading…
1
3
enable mxfp8_cublas recipe in roofline script CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
#1843 opened Mar 5, 2025 by vkuzo Loading…
2
enable torch.compile for mxfp8_cublas recipe CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: performance Use this tag if this PR improves the performance of a feature
#1841 opened Mar 5, 2025 by vkuzo Loading…
3
Implemented CUBLAS-style MXFP scale factor derivation, with test cases.
#1835 opened Mar 4, 2025 by frsun-nvda Loading…
9
Remove libOMP linking for experimental kernels CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: bug fix Use this tag for PRs that fix bugs
#1828 opened Mar 4, 2025 by Jack-Khuu Loading…
3
Add bitcast-flag CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1814 opened Mar 3, 2025 by drisspg Loading…
1
fix USE_CPP=0 during local build CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: bug fix Use this tag for PRs that fix bugs
#1812 opened Mar 3, 2025 by vkuzo Loading…
3
Support MXFP6 packing and fused unpack-dequantise kernel (conflicts resolved) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#1810 opened Mar 3, 2025 by alex-titterton Loading…
18
Refine the document of quantization. CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: documentation Use this tag if this PR adds or improves documentation
#1809 opened Mar 2, 2025 by codereba Loading…
3
Config serde CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: new feature Use this tag if this PR adds a new feature
#1806 opened Mar 1, 2025 by drisspg Loading…
1
[float8] add perf benchmarks for float8 training with rowwise + tensorwise scaling CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: documentation Use this tag if this PR adds or improves documentation
#1793 opened Feb 28, 2025 by danielvegamyhre Loading…
18
Add H100 to CI for regression ci CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: for developers Use this tag if this PR is mainly developer facing
#1792 opened Feb 27, 2025 by jainapurva • Draft
1
Initial stab at string based config parser CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. enhancement New feature or request topic: new feature Use this tag if this PR adds a new feature
#1774 opened Feb 25, 2025 by drisspg • Draft
7
Fix wrong scale eps applied CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. float8 topic: bug fix Use this tag for PRs that fix bugs
#1770 opened Feb 24, 2025 by alexsamardzic Loading…
1
9
Feat: Implementation of the DeepSeek blockwise quantization for fp8 tensors CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1763 opened Feb 22, 2025 by Degnel Loading…
@danielvegamyhre
10
Micro-benchmark inference CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: new feature Use this tag if this PR adds a new feature topic: performance Use this tag if this PR improves the performance of a feature
#1759 opened Feb 21, 2025 by jainapurva Loading…
21
DRAFT: SAM2 bfloat16 (without autocast) and more compile CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1757 opened Feb 21, 2025 by cpuhrsch • Draft
1
ProTip! Updated in the last three days: updated:>2025-03-07.