Skip to content

Commit bbfacb0

Browse files
committed
Update
[ghstack-poisoned]
1 parent 4d13668 commit bbfacb0

File tree

1 file changed

+32
-0
lines changed

1 file changed

+32
-0
lines changed

.github/workflows/pull.yml

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)