Skip to content

[ET-VK] Prevent decomposition of activation ops with native shaders #17361

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
abdelaziz-mahdy wants to merge 6 commits into
base: main
Choose a base branch
from
Open

[ET-VK] Prevent decomposition of activation ops with native shaders #17361

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
0b85421
[ET-VK] Prevent decomposition of activation ops with native shaders
abdelaziz-mahdy Feb 11, 2026
f4ee65d
Merge branch 'main' into vulkan-preserve-activation-ops
SS-JIA Feb 11, 2026
0b8e58c
Revert accidental merge conflict changes in op_registry.py
abdelaziz-mahdy Feb 12, 2026
fe3e20a
Merge branch 'main' into vulkan-preserve-activation-ops
SS-JIA Feb 17, 2026
da38960
Merge branch 'main' of https://github.com/pytorch/executorch into vul…
abdelaziz-mahdy Feb 23, 2026
2159c3e
Merge branch 'vulkan-preserve-activation-ops' of https://github.com/a…
abdelaziz-mahdy Feb 23, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions backends/vulkan/op_registry.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -188,6 +188,8 @@ def register_ephemeral_ops():
exir_ops.edge.aten.exp.default,
exir_ops.edge.aten.gelu.default,
exir_ops.edge.aten.hardshrink.default,
exir_ops.edge.aten.hardsigmoid.default,
exir_ops.edge.aten.hardswish.default,
exir_ops.edge.aten.hardtanh.default,
exir_ops.edge.aten.neg.default,
exir_ops.edge.aten.relu.default,
Expand Down
7 changes: 7 additions & 0 deletions backends/vulkan/partitioner/vulkan_partitioner.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,13 @@
# pyre-ignore
ops_not_to_decompose = [
torch.ops.aten.upsample_nearest2d.vec,
# Activation ops with native Vulkan shaders that PyTorch's default
# decomposition table would otherwise decompose into primitive ops.
# The decomposed paths produce NaN/Inf on some GPUs (e.g. PowerVR)
# due to constant tensor loading issues in the decomposed graph.
torch.ops.aten.hardsigmoid.default,
torch.ops.aten.hardswish.default,
torch.ops.aten.hardshrink.default,
]

logger: logging.Logger = logging.getLogger("")
Expand Down
Loading