diff --git a/backends/vulkan/test/TARGETS b/backends/vulkan/test/TARGETS index 608b3137c36..3287fe79095 100644 --- a/backends/vulkan/test/TARGETS +++ b/backends/vulkan/test/TARGETS @@ -20,6 +20,7 @@ python_unittest( "//executorch/extension/pybindings:portable_lib", # @manual "//executorch/extension/pytree:pylib", "//executorch/kernels/portable:custom_ops_generated_lib", + "//wearables/camera/ml/model_experimentation/person_segmentation/cunet:cunet_lib", ], ) diff --git a/backends/vulkan/test/test_vulkan_delegate.py b/backends/vulkan/test/test_vulkan_delegate.py index dd2142eee47..a4421f0f7c7 100644 --- a/backends/vulkan/test/test_vulkan_delegate.py +++ b/backends/vulkan/test/test_vulkan_delegate.py @@ -17,6 +17,9 @@ from executorch.exir import EdgeProgramManager, to_edge from torch.export import Dim, export, ExportedProgram +from wearables.camera.ml.model_experimentation.person_segmentation.cunet.cunet import ( + C_UNet, +) ctypes.CDLL("libvulkan.so.1") @@ -628,3 +631,13 @@ def forward(self, x): sample_inputs, memory_layouts=[vk_graph_schema.VkMemoryLayout.TENSOR_CHANNELS_PACKED], ) + + def test_vulkan_backend_cunet(self): + module = C_UNet() + sample_inputs = (torch.rand((1, 3, 96, 72), dtype=torch.float32),) + + self.lower_module_and_test_output( + module, + sample_inputs, + memory_layouts=[vk_graph_schema.VkMemoryLayout.TENSOR_CHANNELS_PACKED], + )