Skip to content

Add parameter_name support to _mx_inference_linear_transform#3975

Merged
jcaip merged 2 commits intomainfrom
gh/jcaip/26/head
Mar 5, 2026
Merged

Add parameter_name support to _mx_inference_linear_transform#3975
jcaip merged 2 commits intomainfrom
gh/jcaip/26/head

Conversation

@jcaip
Copy link
Contributor

@jcaip jcaip commented Mar 3, 2026

Stack from ghstack (oldest at bottom):

Add parameter_name keyword-only argument to support FqnToConfig
per-parameter quantization. Replace module.weight reads/writes with
getattr/setattr, and use _module_extra_repr partial pattern.

Add `parameter_name` keyword-only argument to support FqnToConfig
per-parameter quantization. Replace module.weight reads/writes with
getattr/setattr, and use _module_extra_repr partial pattern.

[ghstack-poisoned]
@pytorch-bot
Copy link

pytorch-bot bot commented Mar 3, 2026

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3975

Note: Links to docs will display an error until the docs builds have been completed.

❌ 1 New Failure, 1 Unrelated Failure

As of commit e4bb6ea with merge base cd062f2 (image):

NEW FAILURE - The following job has failed:

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Mar 3, 2026
Add `parameter_name` keyword-only argument to support FqnToConfig
per-parameter quantization. Replace module.weight reads/writes with
getattr/setattr, and use _module_extra_repr partial pattern.

[ghstack-poisoned]
@jcaip jcaip added module: not user facing Use this tag if you don't want this PR to show up in release notes module: inference quantize_ api inference flow labels Mar 3, 2026
@jcaip jcaip changed the base branch from gh/jcaip/26/base to main March 5, 2026 18:57
@jcaip jcaip merged commit 22607b6 into main Mar 5, 2026
33 of 38 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

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 module: not user facing Use this tag if you don't want this PR to show up in release notes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants