executorch/backends/vulkan/tools/gpuinfo/TARGETS at dd9da2133a5c24537c5a7d61ea4af63bbb54adb6 · pytorch/executorch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
load("@fbcode_macros//build_defs:native_rules.bzl", "buck_filegroup")
load("@fbsource//tools/build_defs:fb_xplat_cxx_binary.bzl", "fb_xplat_cxx_binary")
load(
    "@fbsource//tools/build_defs:platform_defs.bzl",
    "ANDROID",
)
load(
    "@fbsource//xplat/executorch/backends/vulkan:targets.bzl",
    "vulkan_spv_shader_lib",
)

oncall("executorch")

buck_filegroup(
    name = "gpuinfo_shaders",
    srcs = glob([
        "glsl/*",
    ]),
    visibility = [
        "PUBLIC",
    ],
)

vulkan_spv_shader_lib(
    name = "gpuinfo_shader_lib",
    is_fbcode = True,
    spv_filegroups = {
        ":gpuinfo_shaders": "glsl",
    },
)

fb_xplat_cxx_binary(
    name = "vulkan_gpuinfo",
    srcs = glob([
        "**/*.cpp",
    ]),
    headers = glob([
        "**/*.h",
    ]),
    header_namespace = "/include",
    include_directories = ["/include"],
    platforms = ANDROID,
    raw_headers = glob([
        "**/*.h",
    ]),
    deps = [
        ":gpuinfo_shader_lib",
        "//executorch/backends/vulkan:vulkan_graph_runtime",
    ],
)