Skip to content

[Torch FX] Compress PT2E Support#3663

Merged
AlexanderDokuchaev merged 150 commits into
openvinotoolkit:developfrom
anzr299:an/fx/compress_pt2e
Dec 5, 2025
Merged

[Torch FX] Compress PT2E Support#3663
AlexanderDokuchaev merged 150 commits into
openvinotoolkit:developfrom
anzr299:an/fx/compress_pt2e

Conversation

@anzr299
Copy link
Copy Markdown
Collaborator

@anzr299 anzr299 commented Sep 22, 2025

Changes

Introduced a new API to offer weights compression algorithm for quantizers defined in torch.ao.
Currently only supports OpenVINO Quantizer.

Reason for changes

To support Quantizers defined in torch ao.

Related tickets

169342

WC Conformance Test #167: https://github.com/openvinotoolkit/nncf/actions/runs/19372182852 - Pass

@anzr299 anzr299 requested a review from a team as a code owner September 22, 2025 14:43
@github-actions github-actions Bot added the API Public API-impacting changes label Sep 22, 2025
@anzr299 anzr299 marked this pull request as draft September 22, 2025 14:56
@daniil-lyakhov daniil-lyakhov self-requested a review September 22, 2025 15:03
Comment thread src/nncf/experimental/torch/fx/quantization/quantize_pt2e.py Outdated
Comment thread src/nncf/experimental/quantization/algorithms/weight_compression/algorithm.py Outdated
Comment thread src/nncf/experimental/torch/fx/quantization/quantizer/__init__.py Outdated
Comment thread tests/executorch/__init__.py Outdated
Comment thread tests/executorch/test_quantizer.py Outdated
Comment thread tests/executorch/observers.py Outdated
Copy link
Copy Markdown
Collaborator

@daniil-lyakhov daniil-lyakhov left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can I see the PR with OpenVINOQuantizer?

Comment thread src/nncf/experimental/quantization/algorithms/weight_compression/algorithm.py Outdated
Comment thread src/nncf/experimental/quantization/algorithms/weight_compression/algorithm.py Outdated
Comment on lines +34 to +35
) -> torch.fx.GraphModule:
self._quantizer = quantizer
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

typehints an docstring are missing

Comment thread src/nncf/quantization/algorithms/weight_compression/algorithm.py Outdated
Comment thread src/nncf/experimental/quantization/algorithms/weight_compression/algorithm.py Outdated
Comment thread src/nncf/experimental/torch/fx/quantization/quantize_pt2e.py
@AlexanderDokuchaev AlexanderDokuchaev merged commit bb510f9 into openvinotoolkit:develop Dec 5, 2025
20 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

API Public API-impacting changes NNCF PT Pull requests that updates NNCF PyTorch

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants