1+ load ("@fbcode_macros//build_defs:build_file_migration.bzl" , "fbcode_target" , "non_fbcode_target" )
12load ("@fbsource//tools/build_defs:fb_native_wrapper.bzl" , "fb_native" )
23load (":compute_api_tests.bzl" , "define_compute_api_test_targets" )
34
4- fb_native .filegroup (
5+ oncall ("executorch" )
6+ non_fbcode_target (_kind = fb_native .filegroup ,
57 name = "test_shaders" ,
68 srcs = glob ([
79 "glsl/*" ,
@@ -11,4 +13,108 @@ fb_native.filegroup(
1113 ],
1214)
1315
14- define_compute_api_test_targets ()
16+ non_fbcode_target (_kind = define_compute_api_test_targets ,)
17+
18+ # !!!! fbcode/executorch/backends/vulkan/test/TARGETS was merged into this file, see https://fburl.com/workplace/xl8l9yuo for more info !!!!
19+
20+ load ("@fbcode_macros//build_defs:python_unittest.bzl" , "python_unittest" )
21+ load ("@fbsource//xplat/executorch/build:runtime_wrapper.bzl" , "runtime" )
22+
23+
24+ fbcode_target (_kind = python_unittest ,
25+ name = "test_vulkan_delegate" ,
26+ srcs = [
27+ "test_vulkan_delegate.py" ,
28+ ],
29+ preload_deps = [
30+ "fbsource//third-party/swiftshader/lib/linux-x64:libvk_swiftshader_fbcode" ,
31+ "//executorch/backends/vulkan:vulkan_backend_lib" ,
32+ "//executorch/kernels/portable:custom_ops_generated_lib" ,
33+ ],
34+ deps = [
35+ ":test_utils" ,
36+ "//caffe2:torch" ,
37+ "//executorch/backends/transforms:convert_dtype_pass" ,
38+ "//executorch/backends/vulkan:vulkan_preprocess" ,
39+ "//executorch/backends/vulkan/partitioner:vulkan_partitioner" ,
40+ "//executorch/exir:lib" ,
41+ "//executorch/extension/pybindings:portable_lib" , # @manual
42+ "//executorch/extension/pytree:pylib" ,
43+ "//executorch/kernels/portable:custom_ops_generated_lib" ,
44+ ],
45+ )
46+
47+ fbcode_target (_kind = python_unittest ,
48+ name = "test_vulkan_passes" ,
49+ srcs = [
50+ "test_vulkan_passes.py" ,
51+ ],
52+ deps = [
53+ "//caffe2:torch" ,
54+ "//executorch/backends/vulkan/_passes:vulkan_passes" ,
55+ "//executorch/backends/vulkan:vulkan_preprocess" ,
56+ "//executorch/backends/xnnpack/quantizer:xnnpack_quantizer" ,
57+ "//pytorch/ao:torchao" , # @manual
58+ ]
59+ )
60+
61+ fbcode_target (_kind = python_unittest ,
62+ name = "test_vulkan_delegate_header" ,
63+ srcs = [
64+ "test_vulkan_delegate_header.py" ,
65+ ],
66+ deps = [
67+ "//executorch/backends/vulkan:vulkan_preprocess" ,
68+ ],
69+ )
70+
71+ fbcode_target (_kind = python_unittest ,
72+ name = "test_serialization" ,
73+ srcs = [
74+ "test_serialization.py" ,
75+ ],
76+ deps = [
77+ "//caffe2:torch" ,
78+ "//executorch/backends/vulkan:vulkan_preprocess" ,
79+ ],
80+ )
81+
82+ fbcode_target (_kind = python_unittest ,
83+ name = "test_vulkan_tensor_repr" ,
84+ srcs = [
85+ "test_vulkan_tensor_repr.py" ,
86+ ],
87+ deps = [
88+ "//caffe2:torch" ,
89+ "//executorch/backends/vulkan:vulkan_preprocess" ,
90+ ],
91+ )
92+
93+ fbcode_target (_kind = runtime .python_library ,
94+ name = "tester" ,
95+ srcs = ["tester.py" ],
96+ deps = [
97+ "//executorch/backends/vulkan/partitioner:vulkan_partitioner" ,
98+ "//executorch/backends/vulkan:vulkan_preprocess" ,
99+ ]
100+ )
101+
102+ fbcode_target (_kind = runtime .python_library ,
103+ name = "test_utils" ,
104+ srcs = [
105+ "utils.py" ,
106+ ],
107+ deps = [
108+ "//caffe2:torch" ,
109+ "//executorch/backends/vulkan:vulkan_preprocess" ,
110+ "//executorch/backends/vulkan/partitioner:vulkan_partitioner" ,
111+ "//executorch/backends/xnnpack:xnnpack_preprocess" ,
112+ "//executorch/backends/xnnpack/quantizer:xnnpack_quantizer" ,
113+ "//executorch/backends/xnnpack/partition:xnnpack_partitioner" ,
114+ "//executorch/devtools:lib" ,
115+ "//executorch/devtools/bundled_program/serialize:lib" ,
116+ "//executorch/exir:lib" ,
117+ "//executorch/extension/pybindings:portable_lib" , # @manual
118+ "//executorch/extension/pytree:pylib" ,
119+ ],
120+ )
0 commit comments