Skip to content

Pull requests: pytorch/ao

New pull request New
114 Open 1,210 Closed
114 Open 1,210 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

[Experimental] Enable kleidi AI examples to run on graviton3
#1721 opened Feb 17, 2025 by akote123 Loading…
2
CPUOffload: only offload parameters above a certain size CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1720 opened Feb 16, 2025 by ngc92 Loading…
1
Make TorchAO cpp/Python extension CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported topic: not user facing Use this tag if you don't want this PR to show up in release notes
#1719 opened Feb 14, 2025 by drisspg Loading…
4
fix tensor parallelism for float8 training with rowwise scaling 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)
#1718 opened Feb 14, 2025 by vkuzo Loading…
1
1
Make FakeQuantizer expose useful config details 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)
#1717 opened Feb 14, 2025 by andrewor14 Loading…
1
Update version.txt to 0.10.0 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
#1714 opened Feb 14, 2025 by HDCharles Loading…
1
SAM2: Use torch.export for VOS CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1708 opened Feb 13, 2025 by cpuhrsch Loading…
1
Outline Benchmark Quant APIs 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 topic: new feature Use this tag if this PR adds a new feature
#1706 opened Feb 12, 2025 by jainapurva • Draft
2
[ROCm] preshuffled weight mm CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: rocm
#1702 opened Feb 11, 2025 by jeffdaily • Draft
2
fix overflow fp16 act woq int8 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1698 opened Feb 11, 2025 by leslie-fang-intel Loading…
1
Fix float8nocompile CI workflow ci 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
#1695 opened Feb 10, 2025 by danielvegamyhre Loading…
1
Support MXFP6 packing and fused unpack-dequantise kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1687 opened Feb 10, 2025 by alex-titterton Loading…
1
Fix DDP with nf4 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
#1684 opened Feb 9, 2025 by jeromeku Loading…
1
2
ROCm OCP FP8 Support CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. float8 module: rocm topic: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#1677 opened Feb 6, 2025 by petrex Loading…
@petrex
3
Add CUTLASS-based row-wise scaled sparse FP8 kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. float8 sparsity topic: new feature Use this tag if this PR adds a new feature
#1671 opened Feb 5, 2025 by alexsamardzic • Draft
4
Feat/blockwise fp8 quant CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1668 opened Feb 5, 2025 by Degnel Loading…
6
Tensor parallel support for fpx dtype CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1660 opened Feb 4, 2025 by jainapurva • Draft
1
Attempt to switch everything to cmake CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1659 opened Feb 4, 2025 by drisspg Loading…
7
Tensor parallel support for uintx dtype 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)
#1656 opened Feb 3, 2025 by jainapurva • Draft
1
draft ukernel selection logic CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1652 opened Feb 3, 2025 by metascroy • Draft
29
[Fix]: Fallback to KleidiAI channelwise kernel groupsize isnt suitable CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1647 opened Jan 31, 2025 by nikhil-arm Loading…
5
Tests jeanschmidt/NVIDIA_IMEX_CHANNELS changes CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1644 opened Jan 30, 2025 by jeanschmidt Loading…
1
Create separate float8 tensor subclass CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1636 opened Jan 29, 2025 by jainapurva • Draft
3
Fix broken nav link in README
#1633 opened Jan 28, 2025 by ByronHsu Loading…
2
[not for land] hook up MX to CUDA 12.8 cuBLAS MX gemm 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
#1625 opened Jan 27, 2025 by vkuzo Loading…
4
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.