Skip to content

Commit 4bcd724

Browse files
committed
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]
2 parents 602d055 + 79073a6 commit 4bcd724

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)