diff --git a/backends/vulkan/op_registry.py b/backends/vulkan/op_registry.py index 6734ada8f98..056869ecc98 100644 --- a/backends/vulkan/op_registry.py +++ b/backends/vulkan/op_registry.py @@ -1356,7 +1356,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 b48d7f98d98..7a18a1282ec 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")