Skip to content

[ET-VK] Migrate to use new q8ta_conv2d ops#17388

Merged
meta-codesync[bot] merged 8 commits intogh/SS-JIA/419/basefrom
gh/SS-JIA/419/head
Feb 13, 2026
Merged

[ET-VK] Migrate to use new q8ta_conv2d ops#17388
meta-codesync[bot] merged 8 commits intogh/SS-JIA/419/basefrom
gh/SS-JIA/419/head

Conversation

@SS-JIA
Copy link
Copy Markdown
Contributor

@SS-JIA SS-JIA commented Feb 11, 2026

Stack from ghstack (oldest at bottom):

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

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](https://our.internmc.facebook.com/intern/diff/D93000168/)

[ghstack-poisoned]
@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented Feb 11, 2026

🔗 Helpful Links

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

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

❌ 1 New Failure

As of commit 4544e52 with merge base dcfd12d (image):

NEW FAILURE - The following job has failed:

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

@github-actions
Copy link
Copy Markdown

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.

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](https://our.internmc.facebook.com/intern/diff/D93000168/)

[ghstack-poisoned]
ssjia and others added 2 commits February 11, 2026 15:13
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](https://our.internmc.facebook.com/intern/diff/D93000168/)

[ghstack-poisoned]
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](https://our.internmc.facebook.com/intern/diff/D93000168/)

[ghstack-poisoned]
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](https://our.internmc.facebook.com/intern/diff/D93000168/)

[ghstack-poisoned]
ssjia added 3 commits February 12, 2026 15:34
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](https://our.internmc.facebook.com/intern/diff/D93000168/)

[ghstack-poisoned]
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](https://our.internmc.facebook.com/intern/diff/D93000168/)

[ghstack-poisoned]
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](https://our.internmc.facebook.com/intern/diff/D93000168/)

[ghstack-poisoned]
@meta-codesync meta-codesync Bot merged commit 469a75e into gh/SS-JIA/419/base Feb 13, 2026
194 of 197 checks passed
@meta-codesync meta-codesync Bot deleted the gh/SS-JIA/419/head branch February 13, 2026 16:13
SS-JIA pushed a commit that referenced this pull request Feb 13, 2026
Pull Request resolved: #17388

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
ghstack-source-id: 340983076
@exported-using-ghexport

Differential Revision: [D93000168](https://our.internmc.facebook.com/intern/diff/D93000168/)
SS-JIA pushed a commit that referenced this pull request Feb 13, 2026
Pull Request resolved: #17388

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
ghstack-source-id: 340983076
@exported-using-ghexport

Differential Revision: [D93000168](https://our.internmc.facebook.com/intern/diff/D93000168/)
SS-JIA pushed a commit that referenced this pull request Feb 13, 2026
Pull Request resolved: #17388

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
ghstack-source-id: 340983076
@exported-using-ghexport

Differential Revision: [D93000168](https://our.internmc.facebook.com/intern/diff/D93000168/)
SS-JIA pushed a commit that referenced this pull request Feb 13, 2026
Pull Request resolved: #17388

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
ghstack-source-id: 340983076
@exported-using-ghexport

Differential Revision: [D93000168](https://our.internmc.facebook.com/intern/diff/D93000168/)
chizkiyahu pushed a commit to chizkiyahu/executorch that referenced this pull request Feb 23, 2026
Pull Request resolved: pytorch#17388

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
ghstack-source-id: 340983076
@exported-using-ghexport

Differential Revision: [D93000168](https://our.internmc.facebook.com/intern/diff/D93000168/)
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. fb-exported meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants