Skip to content

Commit 7a82cbc

Browse files
authored
Updating GPU Arch type
1 parent a8e6db1 commit 7a82cbc

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

Diff for: .github/workflows/run-readme-pr-linuxaarch64.yml

+5-5
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ jobs:
1313
with:
1414
runner: linux.arm64.m7g.4xlarge
1515
docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
16-
gpu-arch-type: cuda
16+
gpu-arch-type: cuda-aarch64
1717
gpu-arch-version: "12.6"
1818
timeout: 60
1919
script: |
@@ -38,7 +38,7 @@ jobs:
3838
with:
3939
runner: linux.arm64.m7g.4xlarge
4040
docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
41-
gpu-arch-type: cuda
41+
gpu-arch-type: cuda-aarch64
4242
gpu-arch-version: "12.6"
4343
timeout: 60
4444
script: |
@@ -58,7 +58,7 @@ jobs:
5858
with:
5959
runner: linux.arm64.m7g.4xlarge
6060
docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
61-
gpu-arch-type: cuda
61+
gpu-arch-type: cuda-aarch64
6262
gpu-arch-version: "12.6"
6363
timeout: 60
6464
script: |
@@ -83,7 +83,7 @@ jobs:
8383
with:
8484
runner: linux.arm64.m7g.4xlarge
8585
docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
86-
gpu-arch-type: cuda
86+
gpu-arch-type: cuda-aarch64
8787
gpu-arch-version: "12.6"
8888
timeout: 60
8989
script: |
@@ -108,7 +108,7 @@ jobs:
108108
with:
109109
runner: linux.arm64.m7g.4xlarge
110110
docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
111-
gpu-arch-type: cuda
111+
gpu-arch-type: cuda-aarch64
112112
gpu-arch-version: "12.6"
113113
timeout: 60
114114
script: |

0 commit comments

Comments
 (0)