diff --git a/backends/vulkan/op_registry.py b/backends/vulkan/op_registry.py index 61135c18648..732cb36f582 100644 --- a/backends/vulkan/op_registry.py +++ b/backends/vulkan/op_registry.py @@ -1391,7 +1391,7 @@ def register_repeat(): @update_features(exir_ops.edge.aten.embedding.default) def register_embedding(): return OpFeatures( - inputs_storage=utils.CHANNELS_PACKED_TEXTURE, + inputs_storage=utils.ANY_STORAGE, inputs_dtypes=[utils.FP_T, utils.INT_T], supports_prepacking=True, supports_resize=True, diff --git a/backends/vulkan/test/op_tests/cases.py b/backends/vulkan/test/op_tests/cases.py index 65ca4206d82..58476b309db 100644 --- a/backends/vulkan/test/op_tests/cases.py +++ b/backends/vulkan/test/op_tests/cases.py @@ -1130,14 +1130,13 @@ def get_embedding_inputs(): Test(weight=[10, 9], indices=[[1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3]]), ] - # Channels packed test cases currently fail on Mac, so they are not included. - # However the test case definition is kept for later debugging. test_suite_cpack = VkTestSuite( [tuple(tc) + (-1, "false", "false") for tc in test_cases] ) test_suite_cpack.dtypes = ["at::kFloat"] test_suite_cpack.layouts = ["utils::kChannelsPacked"] + test_suite_cpack.storage_types = ["utils::kBuffer", "utils::kTexture3D"] test_suite_cpack.test_name_suffix = "cpacked" test_suite_wpack = VkTestSuite( @@ -1149,7 +1148,7 @@ def get_embedding_inputs(): test_suite_wpack.storage_types = ["utils::kBuffer", "utils::kTexture3D"] test_suite_wpack.test_name_suffix = "wpacked" - return test_suite_wpack + return [test_suite_cpack, test_suite_wpack] @register_test_suite("aten.gather.default")