@@ -3,6 +3,8 @@ load("@fbcode_macros//build_defs:python_pytest.bzl", "python_pytest")
33load ("@bazel_skylib//lib:paths.bzl" , "paths" )
44load ("@fbsource//xplat/executorch/build:runtime_wrapper.bzl" , "runtime" )
55
6+ _ENABLE_VGF = True
7+
68def 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