diff --git a/backends/vulkan/op_registry.py b/backends/vulkan/op_registry.py index e3498cf1792..675143cd7fd 100644 --- a/backends/vulkan/op_registry.py +++ b/backends/vulkan/op_registry.py @@ -435,7 +435,19 @@ def register_2d_pool_op(): ) def register_convolution_op(): return OpFeatures( - inputs_storage=utils.CHANNELS_PACKED_TEXTURE, + inputs_storage=[ + utils.CHANNELS_PACKED_TEXTURE, # input + utils.NO_STORAGE, # weight (prepacked) + utils.NO_STORAGE, # bias (prepacked) + utils.NO_STORAGE, # stride (non tensor) + utils.NO_STORAGE, # padding (non tensor) + utils.NO_STORAGE, # dilation (non tensor) + utils.NO_STORAGE, # transposed (non tensor) + utils.NO_STORAGE, # output_padding (non tensor) + utils.NO_STORAGE, # groups (non tensor) + utils.NO_STORAGE, # output_min (non tensor) + utils.NO_STORAGE, # output_max (non tensor) + ], supports_resize=True, supports_prepacking=True, )