Skip to content

Pull requests: pytorch/ao

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
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort

Pull requests list

Update IntxUnpackedTensor to support dynamic activation 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
#2861 opened Aug 22, 2025 by metascroy Loading…
Port metadata from the linear node onto the reference custom op for int4 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. pt2e_quant pt2 export quantization (prepare_pt2e, convert_pt2e)
#2860 opened Aug 22, 2025 by kimishpatel Loading…
Enable quantizing local checkpoints in model release 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
#2859 opened Aug 22, 2025 by jerryzh168 Loading…
[moe training] add test case for shared expert in distributed tests CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2856 opened Aug 22, 2025 by danielvegamyhre Loading…
[reland] Refactor TorchAOBaseTensor for better BC (#2793) ci-no-td 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: improvement Use this tag if this PR is an improvement (doesn't fit into any of the other categories)
#2855 opened Aug 22, 2025 by jerryzh168 Loading…
use gcnArchName to get gpu_arch
#2853 opened Aug 22, 2025 by sidamd Loading…
[mxfp8 moe training] Add mxfp8 to FSDP and TP tests CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2849 opened Aug 22, 2025 by liangel-02 Draft
[mxfp8 moe] add support for fbgemm 2d-3d mx8mx8bf16 grouped gemm CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2848 opened Aug 22, 2025 by danielvegamyhre Loading…
Fix Llama4 example CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2846 opened Aug 22, 2025 by yiliu30 Loading…
[WIP]Add Int4XPUTensorIntZP CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2845 opened Aug 22, 2025 by liangan1 Loading…
test fsdp2 Moe CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2842 opened Aug 21, 2025 by weifengpy Loading…
[test only] testing adding optioanl tensor arg to float8 tensor 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
#2840 opened Aug 21, 2025 by jerryzh168 Loading…
Conditional ROCm kernel build 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 topic: not user facing Use this tag if you don't want this PR to show up in release notes
#2839 opened Aug 21, 2025 by petrex Loading…
Add MyPy support CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2838 opened Aug 21, 2025 by khushi-411 Loading…
[wip] mx: expose a fast path for casting to fp4x2 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
#2832 opened Aug 21, 2025 by vkuzo Loading…
[fp8 blockwise] wrap triton quantization kernels in custom ops for torch.compile compatibility 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
#2829 opened Aug 21, 2025 by danielvegamyhre Loading…
[fp8 blockwise] load 2d chunks for groupwise quant to enable coalesced gmem accesses 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
#2827 opened Aug 21, 2025 by danielvegamyhre Loading…
use shared bench + profile utils in blockwise fwd bwd bench 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
#2826 opened Aug 21, 2025 by danielvegamyhre Loading…
Fix UT assertion error for int8 sdpa fusion 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
#2816 opened Aug 20, 2025 by Valentine233 Loading…
Molly/enable xpu ci CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2814 opened Aug 20, 2025 by DiweiSun Draft
prototype float8 training quantizeable mm module 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)
#2807 opened Aug 19, 2025 by vkuzo Loading…
xpu ut enabling: test_ao_models CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2803 opened Aug 19, 2025 by DiweiSun Draft
xpu ut enablinng: moe_training CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2802 opened Aug 19, 2025 by DiweiSun Draft
Xpu ut/prototype CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2801 opened Aug 19, 2025 by DiweiSun Draft
xpu ut enbaling: test_integration.py CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2800 opened Aug 19, 2025 by DiweiSun Draft
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.