[ET-VK] Add alignment fields to PackedDimInfo for padded size calculation

[SS-JIA]

Stack from ghstack (oldest at bottom):

• [ET-VK][quantization] Add layout-flexible clone for int8x4 tensors #17171
• [ET-VK][qconv] Add layout-flexible impl of quantized depthwise conv2d #17108
• [ET-VK][ez] Implement helper functions to get fastest moving dim #17107
• [ET-VK][quantization] Implement layout-flexible quantize/dequantize operators #17106
• -> [ET-VK] Add alignment fields to PackedDimInfo for padded size calculation #17170
• [ET-VK][testing] Add per-shader timing breakdown to benchmark output #17105

This change introduces separate alignment fields to PackedDimInfo, decoupling
the alignment used for padding tensor dimensions from the block size used for
packing.

Previously, calculate_padded_sizes used packed_dim_block_size and
outer_packed_dim_block_size directly to determine how much to pad tensor
dimensions. This works but limits flexibility - there are scenarios where we
want to pad dimensions to a larger alignment than the block size for
performance reasons, such as ensuring loads are aligned to cache lines or
removing the need for bounds checking in shaders.

The new fields packed_dim_align and outer_packed_dim_align allow specifying
the alignment independently. For now, these are initialized to match the
corresponding block sizes, preserving existing behavior. Future changes can
set larger alignment values when beneficial for performance.

Authored with Claude.

Differential Revision: D92196649

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/17170

• 📄 Preview Python docs built from this PR

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

❌ 3 New Failures

As of commit 7696369 with merge base 593775b ():

NEW FAILURES - The following jobs have failed:

• Lint / android-java-format / linux-job (gh)
  RuntimeError: Command docker exec -t d1a7870bbf8bcc4278afbd5cc522710550723c719ec495bead8475cd5bcc9844 /exec failed with exit code 1
• pull / android / build-llm-demo / linux-job (gh)
  RuntimeError: Command docker exec -t a4249cfe85794cc868002d41357de6eea7a3e9d008134428759d5f250961eeb7 /exec failed with exit code 1
• pull / unittest-arm-backend-with-no-deps (test_pytest_models_tosa) / linux-job (gh)
  RuntimeError: Command docker exec -t 245df76e4a2c387e3dd2d12a40494f6c6d7e47163a4d63936239ec5635d17416 /exec failed with exit code 1

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

[github-actions]

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.