Skip to content

Commit 6708209

Browse files
committed
Add AArch64 unit test workflow
1 parent 348c748 commit 6708209

File tree

1 file changed

+39
-1
lines changed

1 file changed

+39
-1
lines changed

.github/workflows/pull.yml

Lines changed: 39 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,45 @@ jobs:
3434
3535
# Run tests
3636
pytest
37-
37+
38+
# runner: linux.arm64.2xlarge is not working for some reason
39+
# using matrix instead to dynamically select a runner
40+
generate-matrix-linux-aarch64:
41+
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
42+
with:
43+
package-type: wheel
44+
os: linux-aarch64
45+
test-infra-repository: meta-pytorch/tokenizers
46+
test-infra-ref: main
47+
with-cuda: disabled
48+
with-rocm: disabled
49+
build-python-only: enabled
50+
51+
unittest-linux-aarch64:
52+
name: unittest-linux-aarch64
53+
needs: generate-matrix-linux-aarch64
54+
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
55+
strategy:
56+
matrix: ${{ fromJson(needs.generate-matrix-linux-aarch64.outputs.matrix) }}
57+
fail-fast: false
58+
with:
59+
runner: ${{ matrix.validation_runner }}
60+
submodules: 'true'
61+
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
62+
timeout: 90
63+
script: |
64+
set -ex
65+
cmake -DCMAKE_BUILD_TYPE=Debug test -Bbuild/test
66+
cmake --build build/test -j9 --config Debug
67+
pushd build/test && ctest && popd
68+
69+
# Install tokenizers
70+
pip install . -v
71+
pip install pytest blobfile transformers>=4.53.1
72+
73+
# Run tests
74+
pytest
75+
3876
unittest-windows:
3977
uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
4078
with:

0 commit comments

Comments
 (0)