Skip to content

Pull requests: pytorch/ao

New pull request New
327 Open 3,281 Closed
327 Open 3,281 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

fix rocm CI 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: not user facing Use this tag if you don't want this PR to show up in release notes module: rocm
#4437 opened May 22, 2026 by vkuzo Contributor Loading…
2
[CPU] skip registration of PT2E inductor passes if mkldnn not available CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: not user facing Use this tag if you don't want this PR to show up in release notes
#4435 opened May 22, 2026 by Xia-Weiwen Collaborator Loading…
1
2
add fp8 qkv-fusion pass for x86 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4434 opened May 22, 2026 by yuchengliu1 Contributor Loading…
3
Use transpose_result=True for fp8_sparse_mm CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported module: inference quantize_ api inference flow module: rocm
#4433 opened May 21, 2026 by gyllstromk Contributor Loading…
5
fix(pt2e/constant_fold): preserve shared get_attr target when only some uses are dead (#4420) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4431 opened May 20, 2026 by Anai-Guo Contributor Loading…
1 of 2 tasks
1
3
Enable building under a free-threaded Python interpreter CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: core changes affecting multiple modules, e.g. base config/tensor, observers, quant ops
#4428 opened May 19, 2026 by rgommers Contributor Loading…
6
Convert quant_lift_up to CustomGraphPass with uuid() CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: inference quantize_ api inference flow
#4424 opened May 18, 2026 by frgossen Loading…
4
Add RopeSDPAFusionPass(CustomGraphPass) with uuid() CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: inference quantize_ api inference flow
#4423 opened May 18, 2026 by frgossen Loading…
5
Add "32x32 transposed" variant to MXFP8 3D quantization kernel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: training quantize_ api training flow moe mx
#4422 opened May 18, 2026 by alexsamardzic Collaborator Loading…
2
Add _grouped_mm dispatch handler to MXTensor CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: inference quantize_ api inference flow moe mx topic: new feature Use this tag if this PR adds a new feature
#4419 opened May 18, 2026 by xiaowangintel Collaborator Loading…
2
Fix PT2E reference rewrite int kernel dtype mismatch on nightly CPU CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4417 opened May 17, 2026 by Copilot AI • Draft
@xiaowangintel
1
Normalize zero-point dtypes in PT2E reference representation rewrite CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4416 opened May 17, 2026 by Copilot AI • Draft
@xiaowangintel
1
[nvfp4 moe training] Differentiable NVFP4 grouped GEMM for MoE training CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4415 opened May 17, 2026 by roycho96 Contributor Loading…
1
Exclude test/ directory from package distribution CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4414 opened May 16, 2026 by oneraghavan Loading…
2 tasks done
1
1
Guard _is_mkldnn_acl_supported call for non-MKLDNN builds (macOS ARM) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4413 opened May 16, 2026 by oneraghavan Loading…
1 of 2 tasks
1
1
Remove dead setup_fp8_fa3 that imports nonexistent config module CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4412 opened May 16, 2026 by oneraghavan Loading…
3 tasks done
1
1
Convert 4 test files from pytest to unittest (#1621) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4411 opened May 16, 2026 by oneraghavan Loading…
5 tasks done
1
Refactor ToyLinearModel in test_quant_api and test_da8w4_cpu to reuse ToyTwoLinearModel CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4410 opened May 16, 2026 by oneraghavan Loading…
3 tasks done
1
Use transpose_result=True for fp8_sparse_mm (#4409) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported module: rocm
#4409 opened May 15, 2026 by bbeckca Loading…
4
Refactor tests to reuse model architectures from torchao.testing CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4407 opened May 15, 2026 by oneraghavan Loading…
4 tasks done
1
[docs] Resolve doc build warnings CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4406 opened May 15, 2026 by oneraghavan Loading…
3 tasks
1
1
ci: declare workflow-level contents: read on 3 workflows CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4405 opened May 15, 2026 by arpitjain099 Loading…
1
fix(quantizer): guard mkldnn._is_mkldnn_acl_supported for macOS ARM import CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#4404 opened May 15, 2026 by Anai-Guo Contributor Loading…
1 of 3 tasks
1
1
Add algorithm search to FP8 sparse linear benchmark (#4397) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported meta-exported module: rocm
#4397 opened May 13, 2026 by bbeckca Loading…
4
Add _grouped_mm dispatch handler to Float8Tensor CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. float8 module: inference quantize_ api inference flow moe topic: new feature Use this tag if this PR adds a new feature
#4390 opened May 13, 2026 by xiaowangintel Collaborator Loading…
@xiaowangintel
11
ProTip! Updated in the last three days: updated:>2026-05-19.