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