[ET-VK] Migrate to use new q8ta_conv2d ops

[SS-JIA]

Stack from ghstack (oldest at bottom):

• [ET-VK][profiling] Add additional profiling blocks #17394
• [ET-VK][qconv2d][ez] Don't use im2col path for general convs #17393
• [ET-VK] Add nchw_to_int8x4_buffer shader for prepacking int8 staging data #17392
• [ET-VK] Layout-flexible impl of quantized binary #17391
• [ET-VK][ez] Make q8ta_conv2d use 4C1W layout #17390
• [ET-VK][ez] Add AOT support for PackedInt8_4C1W dtype #17389
• -> [ET-VK] Migrate to use new q8ta_conv2d ops #17388
• [ET-VK][qconv] Dynamically select between im2col path and general path #17387
• [ET-VK][qconv][ez] Make q8ta_im2col shader support stride_w != 1 #17386
• Back out "[Diff Train][pytorch/executorch] Apply fixup patch to fbsource" #17399

This renames the quantized convolution custom ops from the old conv2d_q8ta_q8csw_q8to naming convention to the shorter q8ta_conv2d convention, and moves operator registrations from the etvk namespace to the standard et_vk namespace. A new q8ta_conv2d_pw op is also introduced for pointwise (1x1 stride-1 no-padding) convolutions, allowing the pattern matcher to route these to a specialized implementation.

Changes:

• custom_ops_lib.py: Rename functions and op names; add q8ta_conv2d_pw definition
• op_registry.py: Split registration into separate PW and non-PW entries
• quantized_convolution.py: Add pointwise conv detection and routing to q8ta_conv2d_pw
• Q8taConv2d.cpp, Q8taConv2dDW.cpp, Q8taConv2dPW.cpp, Q8taConv2dIm2Col.cpp: Change VK_REGISTER_OP from etvk to et_vk namespace
• TestQ8taConv2d.cpp: Update operator name strings to et_vk namespace

Differential Revision: D93000168

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 2 New Failures, 1 Unrelated Failure

As of commit 34140eb with merge base 964c565 ():

NEW FAILURES - The following jobs have failed:

• Lint / lintrunner / linux-job (gh)
• Test CUDA Builds / export-model-cuda-artifact (openai, whisper-small, quantized-int4-tile-packed) / linux-job (gh)
  RuntimeError: Command docker exec -t 6b0d047f56b2d985702b3e2f06123c535e507c3d90ad07742e85ee68fbdeabbb /exec failed with exit code 1

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

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

• pull / test-samsung-models-linux / linux-job (gh) (trunk failure)
  ##[error]The operation was canceled.

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.