1load("@fbcode_macros//build_defs:python_unittest.bzl", "python_unittest") 2 3oncall("executorch") 4 5python_unittest( 6 name = "test_vulkan_delegate", 7 srcs = [ 8 "test_vulkan_delegate.py", 9 ], 10 preload_deps = [ 11 "fbsource//third-party/swiftshader/lib/linux-x64:libvk_swiftshader_fbcode", 12 "//executorch/backends/vulkan:vulkan_backend_lib", 13 "//executorch/kernels/portable:custom_ops_generated_lib", 14 ], 15 deps = [ 16 "//caffe2:torch", 17 "//executorch/backends/transforms:convert_dtype_pass", 18 "//executorch/backends/vulkan:vulkan_preprocess", 19 "//executorch/backends/vulkan/partitioner:vulkan_partitioner", 20 "//executorch/exir:lib", 21 "//executorch/extension/pybindings:portable_lib", # @manual 22 "//executorch/extension/pytree:pylib", 23 "//executorch/kernels/portable:custom_ops_generated_lib", 24 ], 25) 26 27python_unittest( 28 name = "test_vulkan_delegate_header", 29 srcs = [ 30 "test_vulkan_delegate_header.py", 31 ], 32 deps = [ 33 "//executorch/backends/vulkan:vulkan_preprocess", 34 ], 35) 36 37python_unittest( 38 name = "test_serialization", 39 srcs = [ 40 "test_serialization.py", 41 ], 42 deps = [ 43 "//caffe2:torch", 44 "//executorch/backends/vulkan:vulkan_preprocess", 45 ], 46) 47