File tree Expand file tree Collapse file tree 1 file changed +39
-1
lines changed Expand file tree Collapse file tree 1 file changed +39
-1
lines changed Original file line number Diff line number Diff line change 34
34
35
35
# Run tests
36
36
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
+
38
76
unittest-windows :
39
77
uses : pytorch/test-infra/.github/workflows/windows_job.yml@main
40
78
with :
You can’t perform that action at this time.
0 commit comments