File tree Expand file tree Collapse file tree 1 file changed +32
-0
lines changed
Expand file tree Collapse file tree 1 file changed +32
-0
lines changed Original file line number Diff line number Diff line change @@ -862,6 +862,38 @@ jobs:
862862 PYTHON_EXECUTABLE=python bash examples/nxp/run_aot_example.sh
863863
864864
865+ test-vulkan-models-linux :
866+ name : test-vulkan-models-linux
867+ uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
868+ permissions :
869+ id-token : write
870+ contents : read
871+ with :
872+ runner : linux.2xlarge
873+ docker-image : ci-image:executorch-ubuntu-22.04-clang12
874+ submodules : ' recursive'
875+ ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
876+ timeout : 90
877+ script : |
878+ set -eux
879+
880+ # The generic Linux job chooses to use base env, not the one setup by the image
881+ CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
882+ conda activate "${CONDA_ENV}"
883+
884+ # Setup swiftshader and Vulkan SDK which are required to build the Vulkan delegate
885+ source .ci/scripts/setup-vulkan-linux-deps.sh
886+
887+ # Setup python
888+ PYTHON_EXECUTABLE=python \
889+ CMAKE_ARGS="-DEXECUTORCH_BUILD_VULKAN=ON" \
890+ .ci/scripts/setup-linux.sh --build-tool "cmake"
891+
892+ PYTHON_EXECUTABLE=python bash backends/vulkan/test/scripts/test_model.sh --build
893+
894+ # Test models serially
895+ PYTHON_EXECUTABLE=python bash backends/vulkan/test/scripts/test_model.sh mv2
896+
865897 nxp-build-test :
866898 name : nxp-build-test
867899 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
You can’t perform that action at this time.
0 commit comments