diff --git a/backends/vulkan/op_registry.py b/backends/vulkan/op_registry.py index eed287b3a08..75f8ac331cd 100644 --- a/backends/vulkan/op_registry.py +++ b/backends/vulkan/op_registry.py @@ -1412,6 +1412,7 @@ def register_index_select(): return OpFeatures( inputs_storage=utils.CHANNELS_PACKED_TEXTURE, inputs_dtypes=utils.FP_INT_BOOL_T, + supports_resize=True, ) diff --git a/backends/vulkan/runtime/graph/ops/impl/Embedding.cpp b/backends/vulkan/runtime/graph/ops/impl/Embedding.cpp index 61ba9349b45..80f7b505feb 100644 --- a/backends/vulkan/runtime/graph/ops/impl/Embedding.cpp +++ b/backends/vulkan/runtime/graph/ops/impl/Embedding.cpp @@ -112,7 +112,7 @@ void add_embedding_legacy_node( // Resize Args {}, // Resizing Logic - nullptr)); + resize_embedding_node)); } void embedding(ComputeGraph& graph, const std::vector& args) { diff --git a/backends/vulkan/runtime/graph/ops/impl/IndexSelect.cpp b/backends/vulkan/runtime/graph/ops/impl/IndexSelect.cpp index 20ab76813a7..0c0996a1070 100644 --- a/backends/vulkan/runtime/graph/ops/impl/IndexSelect.cpp +++ b/backends/vulkan/runtime/graph/ops/impl/IndexSelect.cpp @@ -53,7 +53,7 @@ void add_index_select_channel_node( // Resize Args {}, // Resizing Logic - nullptr)); + resize_index_select_channel_node)); } struct IndexSelectParams final { @@ -107,7 +107,7 @@ void add_index_select_node( // Resize Args {}, // Resizing Logic - nullptr)); + resize_fn)); } int64_t get_dim_idx(ComputeGraph& graph, ValueRef in, ValueRef dim_ref) {