File tree Expand file tree Collapse file tree 2 files changed +110
-24
lines changed
Expand file tree Collapse file tree 2 files changed +110
-24
lines changed Original file line number Diff line number Diff line change 1+ name : Build and test Linux CUDA aarch64 wheels
2+
3+ on :
4+ pull_request :
5+ push :
6+ branches :
7+ - nightly
8+ - main
9+ - release/*
10+ tags :
11+ - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
12+ workflow_dispatch :
13+
14+ concurrency :
15+ group : ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
16+ cancel-in-progress : true
17+
18+ permissions :
19+ id-token : write
20+ contents : write
21+
22+ defaults :
23+ run :
24+ shell : bash -l -eo pipefail {0}
25+
26+ jobs :
27+ generate-matrix :
28+ uses : pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
29+ with :
30+ package-type : wheel
31+ os : linux-aarch64
32+ test-infra-repository : pytorch/test-infra
33+ test-infra-ref : main
34+ with-cpu : disable
35+ with-xpu : disable
36+ with-rocm : disable
37+ with-cuda : enable
38+ build-python-only : " disable"
39+ build :
40+ needs : generate-matrix
41+ strategy :
42+ fail-fast : false
43+ name : Build and Upload wheel
44+ uses : pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
45+ with :
46+ repository : meta-pytorch/torchcodec
47+ ref : " "
48+ test-infra-repository : pytorch/test-infra
49+ test-infra-ref : main
50+ build-matrix : ${{ needs.generate-matrix.outputs.matrix }}
51+ pre-script : packaging/pre_build_script.sh
52+ post-script : packaging/post_build_script.sh
53+ smoke-test-script : packaging/fake_smoke_test.py
54+ package-name : torchcodec
55+ trigger-event : ${{ github.event_name }}
56+ architecture : aarch64
57+ build-platform : " python-build-package"
58+ build-command : " BUILD_AGAINST_ALL_FFMPEG_FROM_S3=1 ENABLE_CUDA=1 python -m build --wheel -vvv --no-isolation"
Original file line number Diff line number Diff line change 1515)
1616
1717if (LINUX)
18- set (
19- platform_url
20- ${base_url} /linux_x86_64
21- )
18+ if (CMAKE_SYSTEM_PROCESSOR MATCHES "aarch64|arm64|ARM64" )
19+ set (
20+ platform_url
21+ ${base_url} /linux_aarch64
22+ )
2223
23- set (
24- f4_sha256
25- 1a083f1922443bedb5243d04896383b8c606778a7ddb9d886c8303e55339fe0c
26- )
27- set (
28- f5_sha256
29- 65d6ad54082d94dcb3f801d73df2265e0e1bb303c7afbce7723e3b77ccd0e207
30- )
31- set (
32- f6_sha256
33- 8bd5939c2f4a4b072e837e7870c13fe7d13824e5ff087ab534e4db4e90b7be9c
34- )
35- set (
36- f7_sha256
37- 1cb946d8b7c6393c2c3ebe1f900b8de7a2885fe614c45d4ec32c9833084f2f26
38- )
39- set (
40- f8_sha256
41- c55b3c1a4b5e4d5fdd7c632bea3ab6f45b4e37cc8e0999dda3f84a8ed8defad8
42- )
24+ set (
25+ f4_sha256
26+ a310a2ed9ffe555fd3278dae15065541098dd35e124564671dcda6a6620ac842
27+ )
28+ set (
29+ f5_sha256
30+ 89ca7996bccbc2db49adaa401d20fdbabffe0e1b4e07a0f81d6b143e858b7c8d
31+ )
32+ set (
33+ f6_sha256
34+ ae44c67b4587d061b8e9cc8990ca891ee013fe52ad79e5016ba29871562621da
35+ )
36+ set (
37+ f7_sha256
38+ 948e2cac66ca6f68ff526d5e84138e94bce0f1a7c83f502d15d85d0bd3ddc112
39+ )
40+ set (
41+ f8_sha256
42+ b9cfd99ae75a14e58300854967d4dc49de0b3daa551df51ea1f52a3f08d2c8af
43+ )
44+ elseif (LINUX) # assume x86_64
45+ set (
46+ platform_url
47+ ${base_url} /linux_x86_64
48+ )
49+
50+ set (
51+ f4_sha256
52+ 1a083f1922443bedb5243d04896383b8c606778a7ddb9d886c8303e55339fe0c
53+ )
54+ set (
55+ f5_sha256
56+ 65d6ad54082d94dcb3f801d73df2265e0e1bb303c7afbce7723e3b77ccd0e207
57+ )
58+ set (
59+ f6_sha256
60+ 8bd5939c2f4a4b072e837e7870c13fe7d13824e5ff087ab534e4db4e90b7be9c
61+ )
62+ set (
63+ f7_sha256
64+ 1cb946d8b7c6393c2c3ebe1f900b8de7a2885fe614c45d4ec32c9833084f2f26
65+ )
66+ set (
67+ f8_sha256
68+ c55b3c1a4b5e4d5fdd7c632bea3ab6f45b4e37cc8e0999dda3f84a8ed8defad8
69+ )
70+ endif ()
4371elseif (APPLE )
4472 set (
4573 platform_url
You can’t perform that action at this time.
0 commit comments