[wip] make nvfp4 scale shape match qdata

[vkuzo]

Summary:

maybe 90% there but does not work yet, need to fix tests

we'll need this to stitch MoE weights in vLLM

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

[vkuzo]

Stack from ghstack (oldest at bottom):

• -> [wip] make nvfp4 scale shape match qdata #3094
• mx inference test: fix on B200 #3093

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 7 New Failures

As of commit d37622c with merge base 75944ab ():

NEW FAILURES - The following jobs have failed:

• Code Analysis with Ruff / build (3.9) (gh)
  Process completed with exit code 1.
• PR Label Check / Check PR Labels (gh)
  Process completed with exit code 1.
• Run 1xH100 Tests / test (H100, linux.aws.h100, --pre torch torchvision torchaudio fbgemm-gpu-genai --index-url https... / linux-job (gh)
  RuntimeError: Command docker exec -t 9fea7b4b91317d9f34cf17c02ae3e0d347fd788b76efd6b6c348bff994c88760 /exec failed with exit code 1
• Run Regression Tests / test (CPU 2.8, linux.4xlarge, torch==2.8.0 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  RuntimeError: Command docker exec -t c01815b3afe1b96ded000d66f7eaa4b981c4b28de0e77c9ba3cb19194e51daba /exec failed with exit code 1
• Run Regression Tests / test (CUDA 2.8, linux.g5.12xlarge.nvidia.gpu, torch==2.8.0, cuda, 12.6) / linux-job (gh)
  RuntimeError: Command docker exec -t 8abe2bc4840454bff05de85d9ededcfe0401976bc01737c705259c2a2808c49c /exec failed with exit code 1
• Run Regression Tests / test-nightly (CPU Nightly, linux.4xlarge, --pre torch --index-url https://download.pytorch.org/wh... / linux-job (gh)
  RuntimeError: Command docker exec -t 4d0dbc734c0d3299706096307570122f4a6ee0e98f672f6d76dcc611c5003e06 /exec failed with exit code 1
• Run Regression Tests / test-nightly (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch --index-url https://downloa... / linux-job (gh)
  RuntimeError: Command docker exec -t 049639b60e0bd902624435fcff912147255b151b0fa9e897752939836c6aa419 /exec failed with exit code 1

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