Add aarch64 to validation framework (#1489)

pytorch · Aug 23, 2023 · 3f1e42f · 3f1e42f
1 parent 331b031
commit 3f1e42f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh
@@ -27,7 +27,7 @@ else
 
     if [[ ${TARGET_OS} == 'windows' ]]; then
         python  ./test/smoke_test/smoke_test.py
-    elif [[ ${TARGET_OS} == 'aarch64-linux'  ]]; then
+    elif [[ ${TARGET_OS} == 'linux-aarch64'  ]]; then
         python3  ./test/smoke_test/smoke_test.py  --package=torchonly
     else
         python3  ./test/smoke_test/smoke_test.py

diff --git a/.github/workflows/validate-aarch64-linux-binaries.yml b/.github/workflows/validate-aarch64-linux-binaries.yml
@@ -54,7 +54,7 @@ jobs:
       script: |
         set -ex
         export ENV_NAME="conda-env-${{ github.run_id }}"
-        export TARGET_OS="aarch64-linux"
+        export TARGET_OS="linux-aarch64"
         eval "$(conda shell.bash hook)"
 
         # Standart case: Validate binaries

diff --git a/.github/workflows/validate-binaries.yml b/.github/workflows/validate-binaries.yml
@@ -32,6 +32,7 @@ on:
         options:
           - windows
           - linux
+          - linux-aarch64
           - macos
           - all
       channel:
@@ -65,6 +66,13 @@ jobs:
       channel: ${{ inputs.channel }}
       ref: ${{ inputs.ref || github.ref }}
 
+  linux:
+    if:  inputs.os == 'linux-aarch64'
+    uses: ./.github/workflows/validate-aarch64-linux-binaries.yml
+    with:
+      channel: ${{ inputs.channel }}
+      ref: ${{ inputs.ref || github.ref }}
+
   mac:
     if:  inputs.os == 'macos' || inputs.os == 'all'
     uses: ./.github/workflows/validate-macos-binaries.yml