From 82542937ab061ebc10130cc05f30088e2ed7a725 Mon Sep 17 00:00:00 2001 From: Andrea Nicastro Date: Mon, 2 Jun 2025 00:36:49 -0700 Subject: [PATCH] Add where layer to ops registry Summary: To export models to the ET vulkan backend it is necessary that the op is added to the registry. This also registers the buffer implementation of the binary operations. Differential Revision: D75686562 --- backends/vulkan/op_registry.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/backends/vulkan/op_registry.py b/backends/vulkan/op_registry.py index 0486110ced6..754696f7ac5 100644 --- a/backends/vulkan/op_registry.py +++ b/backends/vulkan/op_registry.py @@ -259,6 +259,17 @@ def register_binary_op(features: OpFeatures): valid_packed_dims=all_packed_dims, ) features.resize_fn = True + features.buffer_impl = True + return features + + +@update_features(exir_ops.edge.aten.where.self) +def register_where_op(features: OpFeatures): + features.texture_impl = TextureImplFeatures( + valid_packed_dims=all_packed_dims, + ) + features.buffer_impl = True + features.resize_fn = True return features