Skip to content

Add parameter_name support to _nvfp4_inference_linear_transform#3976

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

Add parameter_name support to _nvfp4_inference_linear_transform#3976
jcaip merged 5 commits intomainfrom
gh/jcaip/27/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 the dynamic quantization
branch (step=None) to support FqnToConfig per-parameter quantization.
PREPARE and CONVERT branches are left unchanged as they are
Linear-specific observer flow.

jcaip added 2 commits March 3, 2026 07:00
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]
Add `parameter_name` keyword-only argument to the dynamic quantization
branch (step=None) to support FqnToConfig per-parameter quantization.
PREPARE and CONVERT branches are left unchanged as they are
Linear-specific observer flow.

[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/3976

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

✅ You can merge normally! (1 Unrelated Failure)

As of commit c850131 with merge base 22607b6 (image):

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
jcaip added 2 commits March 3, 2026 07:06
…ence_linear_transform"

Add `parameter_name` keyword-only argument to the dynamic quantization
branch (step=None) to support FqnToConfig per-parameter quantization.
PREPARE and CONVERT branches are left unchanged as they are
Linear-specific observer flow.

[ghstack-poisoned]
…sform"

Add `parameter_name` keyword-only argument to the dynamic quantization
branch (step=None) to support FqnToConfig per-parameter quantization.
PREPARE and CONVERT branches are left unchanged as they are
Linear-specific observer flow.

[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/27/base to main March 5, 2026 18:58
@jcaip jcaip merged commit 5a5029d into main Mar 5, 2026
38 of 40 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