[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp - #14415

Merged
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps
Oct 9, 2025
Merged

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp#14415
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps

Conversation

@alexdean08

@alexdean08alexdean08 commented Sep 18, 2025

Copy link
Copy Markdown
Contributor

With the motivation of improving performance, this change adds the functionality for fusing the following ops:

  • conv2d PW s1p0 and binary ops (add, sub, mul, div)
  • clamp and binary ops (add, sub, mul, div)
  • clamp and clamp

cc @SS-JIA@manuelcandales@digantdesai@cbilgin

@alexdean08alexdean08 added module: vulkan Issues related to the Vulkan delegate and code under backends/vulkan/ release notes: vulkan Changes to the Vulkan backend delegate labels Sep 18, 2025
@pytorch-bot

pytorch-botBot commented Sep 18, 2025

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 3 New Failures, 8 Pending

As of commit ecc521f with merge base 2eb8994 (image):

NEW FAILURES - The following jobs have failed:

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

@meta-clameta-claBot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 18, 2025
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

import sys

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@alexdean08 would you mind taking a stab at implementing the fusions via the new patterns/ folder I added? This folder is meant to abstract matching fusable operator patterns and fusing them into a single op.

Here is an example for quantized convolution: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/quantized_convolution.py

Basically, you will need to implement 3 things in a file called, say patterns/fused_convolution.py

# patterns/fused_convolution.pyFUSEABLE_OPS= [
exir_ops.edge.aten.relu.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.clamp.default,
]
FUSEABLE_BINARY_OPS= [
exir_ops.edge.aten.add.Tensor,
exir_ops.edge.aten.sub.Tensor,
exir_ops.edge.aten.mul.Tensor,
exir_ops.edge.aten.div.Tensor,
]
# Implement a Match class to represent the patterns involved in a pattern matchclassFusedConvolutionMatch(PatternMatch):
def__init__(self, conv_node: torch.fx.Node) ->None:
self.anchor_node=conv_node# Store all relevant args here
...
first_user=list(conv_node.users)[0]
iffirst_usernotinFUSABLE_OPS:
returnself.output_node=first_user# Store other relevant args here
...
self.match_found=Trueconvolution_anchor_nodes= {
exir_ops.edge.aten.conv2d.default,
exir_ops.edge.aten.convolution.default,
}
@register_pattern_detector("fused_convolution")deffind_fused_convolution_patterns(
node: torch.fx.Node,
) ->Optional[FusedConvolutionMatch]:
ifnode.targetnotinconvolution_anchor_nodes:
returnNonematched_pattern=FusedConvolutionMatch(node)
ifmatched_pattern.match_found:
returnmatched_patternreturnNone#### Pattern Replacement##FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET= {
exir_ops.edge.aten.add.Tensor: exir_ops.edge.et_vk.conv_with_binary_add.default
...
}
@register_pattern_replacement("fused_convolution")defmake_fused_conv2d_custom_op(
ep: ExportedProgram,
graph_module: torch.fx.GraphModule,
match: QuantizedConvolutionMatch,
):
withgraph_module.graph.inserting_before(match.output_node):
op_target=FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET[match.output_node.target]
custom_op_node=graph_module.graph.create_node(
"call_function",
op_target,
args=(
...
),
)
qconv_node.meta["val"] =match.output_node.meta["val"]
match.output_node.replace_all_uses_with(custom_op_node)

This should simplify the logic a bit and allow us to reduce the amount of transforms we need to maintain. Although I think fuse_clamps.py still needs to be its own pass.

Once you've added the pattern file, remember to import it here: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/__init__.py so that registrations are triggered when importing the module.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for fusing clamps with binaries, how impactful is it? If the impact is not super significant, I think I would prefer to avoid fusing for the sake of simplifying our codebase.

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Runtime side LGTM. Just comments about how op fusions are implemented. Thanks for working on this!

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. All tests are passing.

@SS-JIA
SS-JIA merged commit a5d7e5c into pytorch:mainOct 9, 2025
176 of 180 checks passed
@SS-JIA

Copy link
Copy Markdown
Contributor

@pytorchbot cherry-pick --onto release/1.0 -c fixnewfeature

pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
…lamp (#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
(cherry picked from commit a5d7e5c)
@pytorchbot

Copy link
Copy Markdown
Collaborator

Cherry picking #14415

The cherry pick PR is at #14960 and it is recommended to link a fixnewfeature cherry pick PR with an issue. The following tracker issues are updated:

Details for Dev Infra team Raised by workflow job

@pytorchbotpytorchbot mentioned this pull request Oct 9, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
#15066)
… Clamp/Clamp (#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
…lamp (pytorch#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
pytorch#15066)
… Clamp/Clamp (pytorch#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
pytorch#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
@xuyanwen2012
xuyanwen2012 deleted the FuseOps branch June 30, 2026 21:58
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/nightlyCLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/release notes: vulkanChanges to the Vulkan backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@alexdean08@SS-JIA@pytorchbot@nil-is-all
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '
Skip to content

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp - #14415

Merged
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps
Oct 9, 2025
Merged

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp#14415
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps

Conversation

@alexdean08

@alexdean08alexdean08 commented Sep 18, 2025

Copy link
Copy Markdown
Contributor

With the motivation of improving performance, this change adds the functionality for fusing the following ops:

  • conv2d PW s1p0 and binary ops (add, sub, mul, div)
  • clamp and binary ops (add, sub, mul, div)
  • clamp and clamp

cc @SS-JIA@manuelcandales@digantdesai@cbilgin

@alexdean08alexdean08 added module: vulkan Issues related to the Vulkan delegate and code under backends/vulkan/ release notes: vulkan Changes to the Vulkan backend delegate labels Sep 18, 2025
@pytorch-bot

pytorch-botBot commented Sep 18, 2025

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 3 New Failures, 8 Pending

As of commit ecc521f with merge base 2eb8994 (image):

NEW FAILURES - The following jobs have failed:

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

@meta-clameta-claBot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 18, 2025
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

import sys

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@alexdean08 would you mind taking a stab at implementing the fusions via the new patterns/ folder I added? This folder is meant to abstract matching fusable operator patterns and fusing them into a single op.

Here is an example for quantized convolution: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/quantized_convolution.py

Basically, you will need to implement 3 things in a file called, say patterns/fused_convolution.py

# patterns/fused_convolution.pyFUSEABLE_OPS= [
exir_ops.edge.aten.relu.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.clamp.default,
]
FUSEABLE_BINARY_OPS= [
exir_ops.edge.aten.add.Tensor,
exir_ops.edge.aten.sub.Tensor,
exir_ops.edge.aten.mul.Tensor,
exir_ops.edge.aten.div.Tensor,
]
# Implement a Match class to represent the patterns involved in a pattern matchclassFusedConvolutionMatch(PatternMatch):
def__init__(self, conv_node: torch.fx.Node) ->None:
self.anchor_node=conv_node# Store all relevant args here
...
first_user=list(conv_node.users)[0]
iffirst_usernotinFUSABLE_OPS:
returnself.output_node=first_user# Store other relevant args here
...
self.match_found=Trueconvolution_anchor_nodes= {
exir_ops.edge.aten.conv2d.default,
exir_ops.edge.aten.convolution.default,
}
@register_pattern_detector("fused_convolution")deffind_fused_convolution_patterns(
node: torch.fx.Node,
) ->Optional[FusedConvolutionMatch]:
ifnode.targetnotinconvolution_anchor_nodes:
returnNonematched_pattern=FusedConvolutionMatch(node)
ifmatched_pattern.match_found:
returnmatched_patternreturnNone#### Pattern Replacement##FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET= {
exir_ops.edge.aten.add.Tensor: exir_ops.edge.et_vk.conv_with_binary_add.default
...
}
@register_pattern_replacement("fused_convolution")defmake_fused_conv2d_custom_op(
ep: ExportedProgram,
graph_module: torch.fx.GraphModule,
match: QuantizedConvolutionMatch,
):
withgraph_module.graph.inserting_before(match.output_node):
op_target=FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET[match.output_node.target]
custom_op_node=graph_module.graph.create_node(
"call_function",
op_target,
args=(
...
),
)
qconv_node.meta["val"] =match.output_node.meta["val"]
match.output_node.replace_all_uses_with(custom_op_node)

This should simplify the logic a bit and allow us to reduce the amount of transforms we need to maintain. Although I think fuse_clamps.py still needs to be its own pass.

Once you've added the pattern file, remember to import it here: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/__init__.py so that registrations are triggered when importing the module.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for fusing clamps with binaries, how impactful is it? If the impact is not super significant, I think I would prefer to avoid fusing for the sake of simplifying our codebase.

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Runtime side LGTM. Just comments about how op fusions are implemented. Thanks for working on this!

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. All tests are passing.

@SS-JIA
SS-JIA merged commit a5d7e5c into pytorch:mainOct 9, 2025
176 of 180 checks passed
@SS-JIA

Copy link
Copy Markdown
Contributor

@pytorchbot cherry-pick --onto release/1.0 -c fixnewfeature

pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
…lamp (#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
(cherry picked from commit a5d7e5c)
@pytorchbot

Copy link
Copy Markdown
Collaborator

Cherry picking #14415

The cherry pick PR is at #14960 and it is recommended to link a fixnewfeature cherry pick PR with an issue. The following tracker issues are updated:

Details for Dev Infra team Raised by workflow job

@pytorchbotpytorchbot mentioned this pull request Oct 9, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
#15066)
… Clamp/Clamp (#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
…lamp (pytorch#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
pytorch#15066)
… Clamp/Clamp (pytorch#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
pytorch#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
@xuyanwen2012
xuyanwen2012 deleted the FuseOps branch June 30, 2026 21:58
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/nightlyCLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/release notes: vulkanChanges to the Vulkan backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@alexdean08@SS-JIA@pytorchbot@nil-is-all
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix"); } } catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp - #14415

Merged
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps
Oct 9, 2025
Merged

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp#14415
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps

Conversation

@alexdean08

@alexdean08alexdean08 commented Sep 18, 2025

Copy link
Copy Markdown
Contributor

With the motivation of improving performance, this change adds the functionality for fusing the following ops:

  • conv2d PW s1p0 and binary ops (add, sub, mul, div)
  • clamp and binary ops (add, sub, mul, div)
  • clamp and clamp

cc @SS-JIA@manuelcandales@digantdesai@cbilgin

@alexdean08alexdean08 added module: vulkan Issues related to the Vulkan delegate and code under backends/vulkan/ release notes: vulkan Changes to the Vulkan backend delegate labels Sep 18, 2025
@pytorch-bot

pytorch-botBot commented Sep 18, 2025

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 3 New Failures, 8 Pending

As of commit ecc521f with merge base 2eb8994 (image):

NEW FAILURES - The following jobs have failed:

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

@meta-clameta-claBot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 18, 2025
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

import sys

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@alexdean08 would you mind taking a stab at implementing the fusions via the new patterns/ folder I added? This folder is meant to abstract matching fusable operator patterns and fusing them into a single op.

Here is an example for quantized convolution: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/quantized_convolution.py

Basically, you will need to implement 3 things in a file called, say patterns/fused_convolution.py

# patterns/fused_convolution.pyFUSEABLE_OPS= [
exir_ops.edge.aten.relu.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.clamp.default,
]
FUSEABLE_BINARY_OPS= [
exir_ops.edge.aten.add.Tensor,
exir_ops.edge.aten.sub.Tensor,
exir_ops.edge.aten.mul.Tensor,
exir_ops.edge.aten.div.Tensor,
]
# Implement a Match class to represent the patterns involved in a pattern matchclassFusedConvolutionMatch(PatternMatch):
def__init__(self, conv_node: torch.fx.Node) ->None:
self.anchor_node=conv_node# Store all relevant args here
...
first_user=list(conv_node.users)[0]
iffirst_usernotinFUSABLE_OPS:
returnself.output_node=first_user# Store other relevant args here
...
self.match_found=Trueconvolution_anchor_nodes= {
exir_ops.edge.aten.conv2d.default,
exir_ops.edge.aten.convolution.default,
}
@register_pattern_detector("fused_convolution")deffind_fused_convolution_patterns(
node: torch.fx.Node,
) ->Optional[FusedConvolutionMatch]:
ifnode.targetnotinconvolution_anchor_nodes:
returnNonematched_pattern=FusedConvolutionMatch(node)
ifmatched_pattern.match_found:
returnmatched_patternreturnNone#### Pattern Replacement##FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET= {
exir_ops.edge.aten.add.Tensor: exir_ops.edge.et_vk.conv_with_binary_add.default
...
}
@register_pattern_replacement("fused_convolution")defmake_fused_conv2d_custom_op(
ep: ExportedProgram,
graph_module: torch.fx.GraphModule,
match: QuantizedConvolutionMatch,
):
withgraph_module.graph.inserting_before(match.output_node):
op_target=FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET[match.output_node.target]
custom_op_node=graph_module.graph.create_node(
"call_function",
op_target,
args=(
...
),
)
qconv_node.meta["val"] =match.output_node.meta["val"]
match.output_node.replace_all_uses_with(custom_op_node)

This should simplify the logic a bit and allow us to reduce the amount of transforms we need to maintain. Although I think fuse_clamps.py still needs to be its own pass.

Once you've added the pattern file, remember to import it here: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/__init__.py so that registrations are triggered when importing the module.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for fusing clamps with binaries, how impactful is it? If the impact is not super significant, I think I would prefer to avoid fusing for the sake of simplifying our codebase.

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Runtime side LGTM. Just comments about how op fusions are implemented. Thanks for working on this!

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. All tests are passing.

@SS-JIA
SS-JIA merged commit a5d7e5c into pytorch:mainOct 9, 2025
176 of 180 checks passed
@SS-JIA

Copy link
Copy Markdown
Contributor

@pytorchbot cherry-pick --onto release/1.0 -c fixnewfeature

pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
…lamp (#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
(cherry picked from commit a5d7e5c)
@pytorchbot

Copy link
Copy Markdown
Collaborator

Cherry picking #14415

The cherry pick PR is at #14960 and it is recommended to link a fixnewfeature cherry pick PR with an issue. The following tracker issues are updated:

Details for Dev Infra team Raised by workflow job

@pytorchbotpytorchbot mentioned this pull request Oct 9, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
#15066)
… Clamp/Clamp (#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
…lamp (pytorch#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
pytorch#15066)
… Clamp/Clamp (pytorch#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
pytorch#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
@xuyanwen2012
xuyanwen2012 deleted the FuseOps branch June 30, 2026 21:58
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/nightlyCLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/release notes: vulkanChanges to the Vulkan backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@alexdean08@SS-JIA@pytorchbot@nil-is-all
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Highlight search terms from Google/DuckDuckGo/Bing referrer\n(function() {\n var ref = document.referrer;\n var terms = [];\n \n if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) {\n var url = new URL(ref);\n var q = url.searchParams.get('q') || url.searchParams.get('p');\n if (q) {\n terms = q.split(/\\s+/).filter(function(t) { return t.length > 2; });\n }\n }\n \n if (terms.length === 0) return;\n \n var style = document.createElement('style');\n style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }';\n document.head.appendChild(style);\n \n function highlight(node) {\n if (node.nodeType === 3) { // text node\n var text = node.textContent;\n var found = false;\n terms.forEach(function(term) {\n var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\') + ')', 'gi');\n if (regex.test(text)) {\n found = true;\n var frag = document.createDocumentFragment();\n var parts = text.split(regex);\n parts.forEach(function(part, i) {\n if (i % 2 === 0) {\n frag.appendChild(document.createTextNode(part));\n } else {\n var span = document.createElement('span');\n span.className = 'userscript-highlight';\n span.textContent = part;\n frag.appendChild(span);\n }\n });\n node.parentNode.replaceChild(frag, node);\n }\n });\n } else if (node.nodeType === 1 && node.childNodes) { // element\n var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT'];\n if (!skipTags.includes(node.tagName)) {\n Array.from(node.childNodes).forEach(highlight);\n }\n }\n }\n \n highlight(document.body);\n \n // Re-highlight on dynamic content\n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1 || node.nodeType === 3) highlight(node);\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Highlight Search Terms"); } } catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp - #14415

Merged
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps
Oct 9, 2025
Merged

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp#14415
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps

Conversation

@alexdean08

@alexdean08alexdean08 commented Sep 18, 2025

Copy link
Copy Markdown
Contributor

With the motivation of improving performance, this change adds the functionality for fusing the following ops:

  • conv2d PW s1p0 and binary ops (add, sub, mul, div)
  • clamp and binary ops (add, sub, mul, div)
  • clamp and clamp

cc @SS-JIA@manuelcandales@digantdesai@cbilgin

@alexdean08alexdean08 added module: vulkan Issues related to the Vulkan delegate and code under backends/vulkan/ release notes: vulkan Changes to the Vulkan backend delegate labels Sep 18, 2025
@pytorch-bot

pytorch-botBot commented Sep 18, 2025

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 3 New Failures, 8 Pending

As of commit ecc521f with merge base 2eb8994 (image):

NEW FAILURES - The following jobs have failed:

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

@meta-clameta-claBot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 18, 2025
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

import sys

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@alexdean08 would you mind taking a stab at implementing the fusions via the new patterns/ folder I added? This folder is meant to abstract matching fusable operator patterns and fusing them into a single op.

Here is an example for quantized convolution: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/quantized_convolution.py

Basically, you will need to implement 3 things in a file called, say patterns/fused_convolution.py

# patterns/fused_convolution.pyFUSEABLE_OPS= [
exir_ops.edge.aten.relu.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.clamp.default,
]
FUSEABLE_BINARY_OPS= [
exir_ops.edge.aten.add.Tensor,
exir_ops.edge.aten.sub.Tensor,
exir_ops.edge.aten.mul.Tensor,
exir_ops.edge.aten.div.Tensor,
]
# Implement a Match class to represent the patterns involved in a pattern matchclassFusedConvolutionMatch(PatternMatch):
def__init__(self, conv_node: torch.fx.Node) ->None:
self.anchor_node=conv_node# Store all relevant args here
...
first_user=list(conv_node.users)[0]
iffirst_usernotinFUSABLE_OPS:
returnself.output_node=first_user# Store other relevant args here
...
self.match_found=Trueconvolution_anchor_nodes= {
exir_ops.edge.aten.conv2d.default,
exir_ops.edge.aten.convolution.default,
}
@register_pattern_detector("fused_convolution")deffind_fused_convolution_patterns(
node: torch.fx.Node,
) ->Optional[FusedConvolutionMatch]:
ifnode.targetnotinconvolution_anchor_nodes:
returnNonematched_pattern=FusedConvolutionMatch(node)
ifmatched_pattern.match_found:
returnmatched_patternreturnNone#### Pattern Replacement##FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET= {
exir_ops.edge.aten.add.Tensor: exir_ops.edge.et_vk.conv_with_binary_add.default
...
}
@register_pattern_replacement("fused_convolution")defmake_fused_conv2d_custom_op(
ep: ExportedProgram,
graph_module: torch.fx.GraphModule,
match: QuantizedConvolutionMatch,
):
withgraph_module.graph.inserting_before(match.output_node):
op_target=FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET[match.output_node.target]
custom_op_node=graph_module.graph.create_node(
"call_function",
op_target,
args=(
...
),
)
qconv_node.meta["val"] =match.output_node.meta["val"]
match.output_node.replace_all_uses_with(custom_op_node)

This should simplify the logic a bit and allow us to reduce the amount of transforms we need to maintain. Although I think fuse_clamps.py still needs to be its own pass.

Once you've added the pattern file, remember to import it here: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/__init__.py so that registrations are triggered when importing the module.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for fusing clamps with binaries, how impactful is it? If the impact is not super significant, I think I would prefer to avoid fusing for the sake of simplifying our codebase.

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Runtime side LGTM. Just comments about how op fusions are implemented. Thanks for working on this!

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. All tests are passing.

@SS-JIA
SS-JIA merged commit a5d7e5c into pytorch:mainOct 9, 2025
176 of 180 checks passed
@SS-JIA

Copy link
Copy Markdown
Contributor

@pytorchbot cherry-pick --onto release/1.0 -c fixnewfeature

pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
…lamp (#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
(cherry picked from commit a5d7e5c)
@pytorchbot

Copy link
Copy Markdown
Collaborator

Cherry picking #14415

The cherry pick PR is at #14960 and it is recommended to link a fixnewfeature cherry pick PR with an issue. The following tracker issues are updated:

Details for Dev Infra team Raised by workflow job

@pytorchbotpytorchbot mentioned this pull request Oct 9, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
#15066)
… Clamp/Clamp (#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
…lamp (pytorch#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
pytorch#15066)
… Clamp/Clamp (pytorch#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
pytorch#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
@xuyanwen2012
xuyanwen2012 deleted the FuseOps branch June 30, 2026 21:58
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/nightlyCLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/release notes: vulkanChanges to the Vulkan backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@alexdean08@SS-JIA@pytorchbot@nil-is-all
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Strip utm_, fbclid, gclid, etc. from all links on page\n(function() {\n var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content',\n 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid',\n 'ref', 'ref_src', 'source', 'medium', 'campaign'];\n \n function cleanUrl(url) {\n try {\n var u = new URL(url, window.location.origin);\n var changed = false;\n trackingParams.forEach(function(p) {\n if (u.searchParams.has(p)) {\n u.searchParams.delete(p);\n changed = true;\n }\n });\n return changed ? u.toString() : url;\n } catch (e) {\n return url;\n }\n }\n \n function cleanLinks() {\n document.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n \n cleanLinks();\n \n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1) {\n if (node.tagName === 'A') cleanLinks();\n node.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Remove Tracking Parameters from Links"); } } catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); } })(); (function(){ try { var __m = "youtube.com"; var __re = new RegExp('^' + "youtube\\.com" + '
Skip to content

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp - #14415

Merged
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps
Oct 9, 2025
Merged

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp#14415
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps

Conversation

@alexdean08

@alexdean08alexdean08 commented Sep 18, 2025

Copy link
Copy Markdown
Contributor

With the motivation of improving performance, this change adds the functionality for fusing the following ops:

  • conv2d PW s1p0 and binary ops (add, sub, mul, div)
  • clamp and binary ops (add, sub, mul, div)
  • clamp and clamp

cc @SS-JIA@manuelcandales@digantdesai@cbilgin

@alexdean08alexdean08 added module: vulkan Issues related to the Vulkan delegate and code under backends/vulkan/ release notes: vulkan Changes to the Vulkan backend delegate labels Sep 18, 2025
@pytorch-bot

pytorch-botBot commented Sep 18, 2025

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 3 New Failures, 8 Pending

As of commit ecc521f with merge base 2eb8994 (image):

NEW FAILURES - The following jobs have failed:

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

@meta-clameta-claBot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 18, 2025
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

import sys

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@alexdean08 would you mind taking a stab at implementing the fusions via the new patterns/ folder I added? This folder is meant to abstract matching fusable operator patterns and fusing them into a single op.

Here is an example for quantized convolution: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/quantized_convolution.py

Basically, you will need to implement 3 things in a file called, say patterns/fused_convolution.py

# patterns/fused_convolution.pyFUSEABLE_OPS= [
exir_ops.edge.aten.relu.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.clamp.default,
]
FUSEABLE_BINARY_OPS= [
exir_ops.edge.aten.add.Tensor,
exir_ops.edge.aten.sub.Tensor,
exir_ops.edge.aten.mul.Tensor,
exir_ops.edge.aten.div.Tensor,
]
# Implement a Match class to represent the patterns involved in a pattern matchclassFusedConvolutionMatch(PatternMatch):
def__init__(self, conv_node: torch.fx.Node) ->None:
self.anchor_node=conv_node# Store all relevant args here
...
first_user=list(conv_node.users)[0]
iffirst_usernotinFUSABLE_OPS:
returnself.output_node=first_user# Store other relevant args here
...
self.match_found=Trueconvolution_anchor_nodes= {
exir_ops.edge.aten.conv2d.default,
exir_ops.edge.aten.convolution.default,
}
@register_pattern_detector("fused_convolution")deffind_fused_convolution_patterns(
node: torch.fx.Node,
) ->Optional[FusedConvolutionMatch]:
ifnode.targetnotinconvolution_anchor_nodes:
returnNonematched_pattern=FusedConvolutionMatch(node)
ifmatched_pattern.match_found:
returnmatched_patternreturnNone#### Pattern Replacement##FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET= {
exir_ops.edge.aten.add.Tensor: exir_ops.edge.et_vk.conv_with_binary_add.default
...
}
@register_pattern_replacement("fused_convolution")defmake_fused_conv2d_custom_op(
ep: ExportedProgram,
graph_module: torch.fx.GraphModule,
match: QuantizedConvolutionMatch,
):
withgraph_module.graph.inserting_before(match.output_node):
op_target=FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET[match.output_node.target]
custom_op_node=graph_module.graph.create_node(
"call_function",
op_target,
args=(
...
),
)
qconv_node.meta["val"] =match.output_node.meta["val"]
match.output_node.replace_all_uses_with(custom_op_node)

This should simplify the logic a bit and allow us to reduce the amount of transforms we need to maintain. Although I think fuse_clamps.py still needs to be its own pass.

Once you've added the pattern file, remember to import it here: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/__init__.py so that registrations are triggered when importing the module.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for fusing clamps with binaries, how impactful is it? If the impact is not super significant, I think I would prefer to avoid fusing for the sake of simplifying our codebase.

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Runtime side LGTM. Just comments about how op fusions are implemented. Thanks for working on this!

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. All tests are passing.

@SS-JIA
SS-JIA merged commit a5d7e5c into pytorch:mainOct 9, 2025
176 of 180 checks passed
@SS-JIA

Copy link
Copy Markdown
Contributor

@pytorchbot cherry-pick --onto release/1.0 -c fixnewfeature

pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
…lamp (#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
(cherry picked from commit a5d7e5c)
@pytorchbot

Copy link
Copy Markdown
Collaborator

Cherry picking #14415

The cherry pick PR is at #14960 and it is recommended to link a fixnewfeature cherry pick PR with an issue. The following tracker issues are updated:

Details for Dev Infra team Raised by workflow job

@pytorchbotpytorchbot mentioned this pull request Oct 9, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
#15066)
… Clamp/Clamp (#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
…lamp (pytorch#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
pytorch#15066)
… Clamp/Clamp (pytorch#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
pytorch#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
@xuyanwen2012
xuyanwen2012 deleted the FuseOps branch June 30, 2026 21:58
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/nightlyCLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/release notes: vulkanChanges to the Vulkan backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@alexdean08@SS-JIA@pytorchbot@nil-is-all
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default"); } } catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp - #14415

Merged
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps
Oct 9, 2025
Merged

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp#14415
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps

Conversation

@alexdean08

@alexdean08alexdean08 commented Sep 18, 2025

Copy link
Copy Markdown
Contributor

With the motivation of improving performance, this change adds the functionality for fusing the following ops:

  • conv2d PW s1p0 and binary ops (add, sub, mul, div)
  • clamp and binary ops (add, sub, mul, div)
  • clamp and clamp

cc @SS-JIA@manuelcandales@digantdesai@cbilgin

@alexdean08alexdean08 added module: vulkan Issues related to the Vulkan delegate and code under backends/vulkan/ release notes: vulkan Changes to the Vulkan backend delegate labels Sep 18, 2025
@pytorch-bot

pytorch-botBot commented Sep 18, 2025

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 3 New Failures, 8 Pending

As of commit ecc521f with merge base 2eb8994 (image):

NEW FAILURES - The following jobs have failed:

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

@meta-clameta-claBot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 18, 2025
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

import sys

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@alexdean08 would you mind taking a stab at implementing the fusions via the new patterns/ folder I added? This folder is meant to abstract matching fusable operator patterns and fusing them into a single op.

Here is an example for quantized convolution: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/quantized_convolution.py

Basically, you will need to implement 3 things in a file called, say patterns/fused_convolution.py

# patterns/fused_convolution.pyFUSEABLE_OPS= [
exir_ops.edge.aten.relu.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.clamp.default,
]
FUSEABLE_BINARY_OPS= [
exir_ops.edge.aten.add.Tensor,
exir_ops.edge.aten.sub.Tensor,
exir_ops.edge.aten.mul.Tensor,
exir_ops.edge.aten.div.Tensor,
]
# Implement a Match class to represent the patterns involved in a pattern matchclassFusedConvolutionMatch(PatternMatch):
def__init__(self, conv_node: torch.fx.Node) ->None:
self.anchor_node=conv_node# Store all relevant args here
...
first_user=list(conv_node.users)[0]
iffirst_usernotinFUSABLE_OPS:
returnself.output_node=first_user# Store other relevant args here
...
self.match_found=Trueconvolution_anchor_nodes= {
exir_ops.edge.aten.conv2d.default,
exir_ops.edge.aten.convolution.default,
}
@register_pattern_detector("fused_convolution")deffind_fused_convolution_patterns(
node: torch.fx.Node,
) ->Optional[FusedConvolutionMatch]:
ifnode.targetnotinconvolution_anchor_nodes:
returnNonematched_pattern=FusedConvolutionMatch(node)
ifmatched_pattern.match_found:
returnmatched_patternreturnNone#### Pattern Replacement##FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET= {
exir_ops.edge.aten.add.Tensor: exir_ops.edge.et_vk.conv_with_binary_add.default
...
}
@register_pattern_replacement("fused_convolution")defmake_fused_conv2d_custom_op(
ep: ExportedProgram,
graph_module: torch.fx.GraphModule,
match: QuantizedConvolutionMatch,
):
withgraph_module.graph.inserting_before(match.output_node):
op_target=FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET[match.output_node.target]
custom_op_node=graph_module.graph.create_node(
"call_function",
op_target,
args=(
...
),
)
qconv_node.meta["val"] =match.output_node.meta["val"]
match.output_node.replace_all_uses_with(custom_op_node)

This should simplify the logic a bit and allow us to reduce the amount of transforms we need to maintain. Although I think fuse_clamps.py still needs to be its own pass.

Once you've added the pattern file, remember to import it here: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/__init__.py so that registrations are triggered when importing the module.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for fusing clamps with binaries, how impactful is it? If the impact is not super significant, I think I would prefer to avoid fusing for the sake of simplifying our codebase.

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Runtime side LGTM. Just comments about how op fusions are implemented. Thanks for working on this!

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. All tests are passing.

@SS-JIA
SS-JIA merged commit a5d7e5c into pytorch:mainOct 9, 2025
176 of 180 checks passed
@SS-JIA

Copy link
Copy Markdown
Contributor

@pytorchbot cherry-pick --onto release/1.0 -c fixnewfeature

pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
…lamp (#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
(cherry picked from commit a5d7e5c)
@pytorchbot

Copy link
Copy Markdown
Collaborator

Cherry picking #14415

The cherry pick PR is at #14960 and it is recommended to link a fixnewfeature cherry pick PR with an issue. The following tracker issues are updated:

Details for Dev Infra team Raised by workflow job

@pytorchbotpytorchbot mentioned this pull request Oct 9, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
#15066)
… Clamp/Clamp (#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
…lamp (pytorch#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
pytorch#15066)
… Clamp/Clamp (pytorch#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
pytorch#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
@xuyanwen2012
xuyanwen2012 deleted the FuseOps branch June 30, 2026 21:58
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/nightlyCLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/release notes: vulkanChanges to the Vulkan backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@alexdean08@SS-JIA@pytorchbot@nil-is-all
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers"); } } catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp - #14415

Merged
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps
Oct 9, 2025
Merged

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp#14415
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps

Conversation

@alexdean08

@alexdean08alexdean08 commented Sep 18, 2025

Copy link
Copy Markdown
Contributor

With the motivation of improving performance, this change adds the functionality for fusing the following ops:

  • conv2d PW s1p0 and binary ops (add, sub, mul, div)
  • clamp and binary ops (add, sub, mul, div)
  • clamp and clamp

cc @SS-JIA@manuelcandales@digantdesai@cbilgin

@alexdean08alexdean08 added module: vulkan Issues related to the Vulkan delegate and code under backends/vulkan/ release notes: vulkan Changes to the Vulkan backend delegate labels Sep 18, 2025
@pytorch-bot

pytorch-botBot commented Sep 18, 2025

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 3 New Failures, 8 Pending

As of commit ecc521f with merge base 2eb8994 (image):

NEW FAILURES - The following jobs have failed:

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

@meta-clameta-claBot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 18, 2025
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

import sys

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@alexdean08 would you mind taking a stab at implementing the fusions via the new patterns/ folder I added? This folder is meant to abstract matching fusable operator patterns and fusing them into a single op.

Here is an example for quantized convolution: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/quantized_convolution.py

Basically, you will need to implement 3 things in a file called, say patterns/fused_convolution.py

# patterns/fused_convolution.pyFUSEABLE_OPS= [
exir_ops.edge.aten.relu.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.clamp.default,
]
FUSEABLE_BINARY_OPS= [
exir_ops.edge.aten.add.Tensor,
exir_ops.edge.aten.sub.Tensor,
exir_ops.edge.aten.mul.Tensor,
exir_ops.edge.aten.div.Tensor,
]
# Implement a Match class to represent the patterns involved in a pattern matchclassFusedConvolutionMatch(PatternMatch):
def__init__(self, conv_node: torch.fx.Node) ->None:
self.anchor_node=conv_node# Store all relevant args here
...
first_user=list(conv_node.users)[0]
iffirst_usernotinFUSABLE_OPS:
returnself.output_node=first_user# Store other relevant args here
...
self.match_found=Trueconvolution_anchor_nodes= {
exir_ops.edge.aten.conv2d.default,
exir_ops.edge.aten.convolution.default,
}
@register_pattern_detector("fused_convolution")deffind_fused_convolution_patterns(
node: torch.fx.Node,
) ->Optional[FusedConvolutionMatch]:
ifnode.targetnotinconvolution_anchor_nodes:
returnNonematched_pattern=FusedConvolutionMatch(node)
ifmatched_pattern.match_found:
returnmatched_patternreturnNone#### Pattern Replacement##FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET= {
exir_ops.edge.aten.add.Tensor: exir_ops.edge.et_vk.conv_with_binary_add.default
...
}
@register_pattern_replacement("fused_convolution")defmake_fused_conv2d_custom_op(
ep: ExportedProgram,
graph_module: torch.fx.GraphModule,
match: QuantizedConvolutionMatch,
):
withgraph_module.graph.inserting_before(match.output_node):
op_target=FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET[match.output_node.target]
custom_op_node=graph_module.graph.create_node(
"call_function",
op_target,
args=(
...
),
)
qconv_node.meta["val"] =match.output_node.meta["val"]
match.output_node.replace_all_uses_with(custom_op_node)

This should simplify the logic a bit and allow us to reduce the amount of transforms we need to maintain. Although I think fuse_clamps.py still needs to be its own pass.

Once you've added the pattern file, remember to import it here: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/__init__.py so that registrations are triggered when importing the module.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for fusing clamps with binaries, how impactful is it? If the impact is not super significant, I think I would prefer to avoid fusing for the sake of simplifying our codebase.

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Runtime side LGTM. Just comments about how op fusions are implemented. Thanks for working on this!

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. All tests are passing.

@SS-JIA
SS-JIA merged commit a5d7e5c into pytorch:mainOct 9, 2025
176 of 180 checks passed
@SS-JIA

Copy link
Copy Markdown
Contributor

@pytorchbot cherry-pick --onto release/1.0 -c fixnewfeature

pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
…lamp (#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
(cherry picked from commit a5d7e5c)
@pytorchbot

Copy link
Copy Markdown
Collaborator

Cherry picking #14415

The cherry pick PR is at #14960 and it is recommended to link a fixnewfeature cherry pick PR with an issue. The following tracker issues are updated:

Details for Dev Infra team Raised by workflow job

@pytorchbotpytorchbot mentioned this pull request Oct 9, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
#15066)
… Clamp/Clamp (#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
…lamp (pytorch#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
pytorch#15066)
… Clamp/Clamp (pytorch#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
pytorch#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
@xuyanwen2012
xuyanwen2012 deleted the FuseOps branch June 30, 2026 21:58
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/nightlyCLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/release notes: vulkanChanges to the Vulkan backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@alexdean08@SS-JIA@pytorchbot@nil-is-all
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Universal Dark Mode - works on any site\n(function() {\n var enabled = true;\n \n function applyDarkMode() {\n if (!enabled) return;\n \n // Create style element if it doesn't exist\n var style = document.getElementById('universal-dark-mode-style');\n if (!style) {\n style = document.createElement('style');\n style.id = 'universal-dark-mode-style';\n document.head.appendChild(style);\n }\n \n // Dark mode CSS - inverts colors but preserves images/video\n style.textContent = '\n /* Invert everything except media */\n html {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #1a1a2e !important;\n }\n \n /* Restore images, videos, iframes, canvas */\n img, video, iframe, canvas, svg, picture, [style*=\"background-image\"] {\n filter: invert(1) hue-rotate(180deg) !important;\n }\n \n /* Preserve specific elements that should not be inverted */\n .no-dark-mode, .no-dark-mode *,\n [data-theme=\"light\"], [data-theme=\"light\"],\n .ace_editor, .ace_editor *,\n .CodeMirror, .CodeMirror *,\n .monaco-editor, .monaco-editor *,\n .markdown-body pre, .markdown-body pre *,\n .highlight, .highlight *,\n pre code, pre code * {\n filter: none !important;\n }\n \n /* Fix common UI elements */\n .modal, .popup, .dropdown-menu, .tooltip, .popover {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #2d2d44 !important;\n border-color: #444 !important;\n }\n \n /* Scrollbars */\n ::-webkit-scrollbar { background: #1a1a2e !important; }\n ::-webkit-scrollbar-thumb { background: #444 !important; }\n ::-webkit-scrollbar-thumb:hover { background: #555 !important; }\n \n /* Selection */\n ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ';\n }\n \n function removeDarkMode() {\n var style = document.getElementById('universal-dark-mode-style');\n if (style) style.remove();\n }\n \n // Toggle with Alt+Shift+D\n document.addEventListener('keydown', function(e) {\n if (e.altKey && e.shiftKey && e.key === 'D') {\n e.preventDefault();\n enabled = !enabled;\n if (enabled) {\n applyDarkMode();\n console.log('[Universal Dark Mode] Enabled');\n } else {\n removeDarkMode();\n console.log('[Universal Dark Mode] Disabled');\n }\n }\n });\n \n // Apply on load\n applyDarkMode();\n \n // Re-apply on dynamic content\n var observer = new MutationObserver(function(mutations) {\n if (enabled && !document.getElementById('universal-dark-mode-style')) {\n applyDarkMode();\n }\n });\n observer.observe(document.head, { childList: true });\n \n console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle');\n})();", "Universal Dark Mode"); } } catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); } })(); })();
Skip to content

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp - #14415

Merged
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps
Oct 9, 2025
Merged

[ET-VK] Add Fusing for Conv/Binary Ops, Clamp/Binary Ops, and Clamp/Clamp#14415
SS-JIA merged 10 commits into
pytorch:mainfrom
sarc-acl:FuseOps

Conversation

@alexdean08

@alexdean08alexdean08 commented Sep 18, 2025

Copy link
Copy Markdown
Contributor

With the motivation of improving performance, this change adds the functionality for fusing the following ops:

  • conv2d PW s1p0 and binary ops (add, sub, mul, div)
  • clamp and binary ops (add, sub, mul, div)
  • clamp and clamp

cc @SS-JIA@manuelcandales@digantdesai@cbilgin

@alexdean08alexdean08 added module: vulkan Issues related to the Vulkan delegate and code under backends/vulkan/ release notes: vulkan Changes to the Vulkan backend delegate labels Sep 18, 2025
@pytorch-bot

pytorch-botBot commented Sep 18, 2025

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 3 New Failures, 8 Pending

As of commit ecc521f with merge base 2eb8994 (image):

NEW FAILURES - The following jobs have failed:

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

@meta-clameta-claBot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 18, 2025
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

import sys

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@alexdean08 would you mind taking a stab at implementing the fusions via the new patterns/ folder I added? This folder is meant to abstract matching fusable operator patterns and fusing them into a single op.

Here is an example for quantized convolution: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/quantized_convolution.py

Basically, you will need to implement 3 things in a file called, say patterns/fused_convolution.py

# patterns/fused_convolution.pyFUSEABLE_OPS= [
exir_ops.edge.aten.relu.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.clamp.default,
]
FUSEABLE_BINARY_OPS= [
exir_ops.edge.aten.add.Tensor,
exir_ops.edge.aten.sub.Tensor,
exir_ops.edge.aten.mul.Tensor,
exir_ops.edge.aten.div.Tensor,
]
# Implement a Match class to represent the patterns involved in a pattern matchclassFusedConvolutionMatch(PatternMatch):
def__init__(self, conv_node: torch.fx.Node) ->None:
self.anchor_node=conv_node# Store all relevant args here
...
first_user=list(conv_node.users)[0]
iffirst_usernotinFUSABLE_OPS:
returnself.output_node=first_user# Store other relevant args here
...
self.match_found=Trueconvolution_anchor_nodes= {
exir_ops.edge.aten.conv2d.default,
exir_ops.edge.aten.convolution.default,
}
@register_pattern_detector("fused_convolution")deffind_fused_convolution_patterns(
node: torch.fx.Node,
) ->Optional[FusedConvolutionMatch]:
ifnode.targetnotinconvolution_anchor_nodes:
returnNonematched_pattern=FusedConvolutionMatch(node)
ifmatched_pattern.match_found:
returnmatched_patternreturnNone#### Pattern Replacement##FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET= {
exir_ops.edge.aten.add.Tensor: exir_ops.edge.et_vk.conv_with_binary_add.default
...
}
@register_pattern_replacement("fused_convolution")defmake_fused_conv2d_custom_op(
ep: ExportedProgram,
graph_module: torch.fx.GraphModule,
match: QuantizedConvolutionMatch,
):
withgraph_module.graph.inserting_before(match.output_node):
op_target=FUSED_NODE_TARGET_TO_CUSTOM_OP_TARGET[match.output_node.target]
custom_op_node=graph_module.graph.create_node(
"call_function",
op_target,
args=(
...
),
)
qconv_node.meta["val"] =match.output_node.meta["val"]
match.output_node.replace_all_uses_with(custom_op_node)

This should simplify the logic a bit and allow us to reduce the amount of transforms we need to maintain. Although I think fuse_clamps.py still needs to be its own pass.

Once you've added the pattern file, remember to import it here: https://github.com/pytorch/executorch/blob/main/backends/vulkan/patterns/__init__.py so that registrations are triggered when importing the module.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also for fusing clamps with binaries, how impactful is it? If the impact is not super significant, I think I would prefer to avoid fusing for the sake of simplifying our codebase.

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Runtime side LGTM. Just comments about how op fusions are implemented. Thanks for working on this!

@SS-JIASS-JIA left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. All tests are passing.

@SS-JIA
SS-JIA merged commit a5d7e5c into pytorch:mainOct 9, 2025
176 of 180 checks passed
@SS-JIA

Copy link
Copy Markdown
Contributor

@pytorchbot cherry-pick --onto release/1.0 -c fixnewfeature

pytorchbot pushed a commit that referenced this pull request Oct 9, 2025
…lamp (#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
(cherry picked from commit a5d7e5c)
@pytorchbot

Copy link
Copy Markdown
Collaborator

Cherry picking #14415

The cherry pick PR is at #14960 and it is recommended to link a fixnewfeature cherry pick PR with an issue. The following tracker issues are updated:

Details for Dev Infra team Raised by workflow job

@pytorchbotpytorchbot mentioned this pull request Oct 9, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
JacobSzwejbka added a commit that referenced this pull request Oct 13, 2025
#15066)
… Clamp/Clamp (#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
…lamp (pytorch#14415)
With the motivation of improving performance, this change adds the
functionality for fusing the following ops:
- conv2d PW s1p0 and binary ops (add, sub, mul, div)
- clamp and binary ops (add, sub, mul, div)
- clamp and clamp
cc @SS-JIA@manuelcandales@digantdesai@cbilgin
jirioc pushed a commit to nxp-upstream/executorch that referenced this pull request Dec 19, 2025
pytorch#15066)
… Clamp/Clamp (pytorch#14415)"
This reverts commit a5d7e5c.
Broke internal builds @SS-JIA is trying to fix this in
pytorch#15058 will leave relanding to
him
### Summary
[PLEASE REMOVE] See [CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests)
for ExecuTorch PR guidelines.
[PLEASE REMOVE] If this PR closes an issue, please add a `Fixes
#<issue-id>` line.
[PLEASE REMOVE] If this PR introduces a fix or feature that should be
the upcoming release notes, please add a "Release notes: <area>" label.
For a list of available release notes labels, check out
[CONTRIBUTING.md's Pull
Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests).
### Test plan
[PLEASE REMOVE] How did you test this PR? Please write down any manual
commands you used and note down tests that you have written if
applicable.
@xuyanwen2012
xuyanwen2012 deleted the FuseOps branch June 30, 2026 21:58
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/nightlyCLA SignedThis label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.module: vulkanIssues related to the Vulkan delegate and code under backends/vulkan/release notes: vulkanChanges to the Vulkan backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@alexdean08@SS-JIA@pytorchbot@nil-is-all