Skip to content

Commit 79073a6

Browse files
committed
Update base for Update on "[ET-VK][ez] Support exporting of custom operator calls via higher_order_auto_functionalized"
As title. This diff adds the ability to partition custom op calls to the Vulkan delegate. Differential Revision: [D63913434](https://our.internmc.facebook.com/intern/diff/D63913434/) [ghstack-poisoned]
1 parent 6a6f3c7 commit 79073a6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

backends/vulkan/partitioner/vulkan_partitioner.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,7 @@ def is_in_local_scalar_dense_chain(self, node: torch.fx.Node) -> bool:
121121
if node.target == exir_ops.edge.aten.select_copy.int:
122122
if len(node.users) != 1:
123123
return False
124+
# pyre-ignore
124125
if node.args[0].meta["val"].numel() != 1:
125126
return False
126127

0 commit comments

Comments
 (0)