[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

[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 51523da with merge base 9368b28 ():

NEW FAILURES - The following jobs have failed:

• Code Analysis with Ruff / build (3.9) (gh)
  test/prototype/mx_formats/test_nvfp4_tensor.py:196:9: F841 Local variable sliced_manually is assigned to but never used
• 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 8809528669dda75f3ba585033b366feaca0df7ba3fb026a8707f612129fd5c2e /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 f8dc573e8dd26b286133e71ed11efa79ec5bc9e12c5914d7ef8427e8528029be /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 855325c7f0aab1e17eeecaa50737717dcc56f5f43f1106192321461c801054d2 /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 585f93af405a6b29bdb109a4ba9a81de400b2fddd8caa37444a3868fe2c8d5e2 /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 3c20e38b89cd8b9a23f449c6c2a7b031ac1299c6367fdeb5ca5f98f2c39e1762 /exec failed with exit code 1

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