Skip to content

Commit c7f1d72

Browse files
authored
Add _ENABLE_VGF toggle to test targets
Differential Revision: D99479233 Pull Request resolved: #18720
1 parent e0e10cc commit c7f1d72

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

backends/arm/test/targets.bzl

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@ load("@fbcode_macros//build_defs:python_pytest.bzl", "python_pytest")
33
load("@bazel_skylib//lib:paths.bzl", "paths")
44
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
55

6+
_ENABLE_VGF = True
7+
68
def define_arm_tests():
79
# TODO [fbonly] Add more tests
810
test_files = []
@@ -67,18 +69,18 @@ def define_arm_tests():
6769
resources = ["conftest.py"],
6870
compile = "with-source",
6971
typing = False,
70-
env = {} if runtime.is_oss else {
72+
env = {} if runtime.is_oss else ({
7173
"MODEL_CONVERTER_PATH": "$(location fbsource//third-party/pypi/ai-ml-sdk-model-converter/0.8.0:model-converter-bin)",
7274
"MODEL_CONVERTER_LIB_DIR": "$(location fbsource//third-party/nvidia-nsight-systems:linux-x86_64)/host-linux-x64",
7375
"LAVAPIPE_LIB_PATH": "$(location fbsource//third-party/mesa/src/gallium/frontends/lavapipe:vulkan_lvp)",
7476
"EMULATION_LAYER_TENSOR_SO": "$(location fbsource//third-party/arm-ml-emulation-layer/v0.9.0/src:libVkLayer_Tensor)",
7577
"EMULATION_LAYER_GRAPH_SO": "$(location fbsource//third-party/arm-ml-emulation-layer/v0.9.0/src:libVkLayer_Graph)",
7678
"EMULATION_LAYER_TENSOR_JSON": "$(location fbsource//third-party/arm-ml-emulation-layer/v0.9.0/src:VkLayer_Tensor_json)",
7779
"EMULATION_LAYER_GRAPH_JSON": "$(location fbsource//third-party/arm-ml-emulation-layer/v0.9.0/src:VkLayer_Graph_json)",
78-
},
80+
} if _ENABLE_VGF else {}),
7981
preload_deps = [
8082
"//executorch/kernels/quantized:custom_ops_generated_lib",
81-
] + ([] if runtime.is_oss else [
83+
] + ([] if runtime.is_oss or not _ENABLE_VGF else [
8284
"fbsource//third-party/khronos:vulkan",
8385
"//executorch/backends/arm/runtime:vgf_backend",
8486
]),

0 commit comments

Comments
 (0)