Add AArch64 unit test workflow

[chizkiyahu]

add tests to check the fails in pytorch/executorch#13946

[chizkiyahu]

@mergennachin thanks
please approve the workflow

[chizkiyahu]

@mergennachin thanks
please approve the workflow

[chizkiyahu]

@larryliu0820

Thanks for approving this.

Do you have an update on when it will be merged?
Should we ping anyone if it needs additional approvals?

[aditew01]

@GregoryComer @larryliu0820 @mergennachin can this be merged, i don't see a mergebot to rebase or trigger a merge

[kirklandsign]

Hi @chizkiyahu I want to avoid duplicated code. How about

  diff --git a/.github/workflows/pull.yml b/.github/workflows/pull.yml
  index dee3f7e..XXXXXXX 100644
  --- a/.github/workflows/pull.yml
  +++ b/.github/workflows/pull.yml
  @@ -13,17 +13,32 @@ concurrency:

   jobs:
     unittest-linux:
  -    name: unittest-linux
  +    name: unittest-linux (${{ matrix.arch }})
       uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
       strategy:
         fail-fast: false
  +      matrix:
  +        include:
  +          - arch: x86_64
  +            runner: linux.2xlarge
  +          - arch: aarch64
  +            runner: linux.arm64.2xlarge
       with:
  -      runner: linux.2xlarge
  +      runner: ${{ matrix.runner }}
         submodules: 'true'
         ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
         timeout: 90
         script: |
           set -ex
  +
  +        # Workaround: on aarch64 the sentencepiece CMakeLists links bare
  +        # "atomic", which some toolchains can't resolve. Substitute the
  +        # variable that holds the full libatomic path instead.
  +        # TODO: drop once sentencepiece upstream handles this.
  +        if [ "${{ matrix.arch }}" = "aarch64" ]; then
  +          sed -i 's/list(APPEND SPM_LIBS "atomic")/list(APPEND SPM_LIBS ${ATOMIC_LIB})/' \
  +            third-party/sentencepiece/src/CMakeLists.txt
  +        fi
  +
           cmake -DCMAKE_BUILD_TYPE=Debug test -Bbuild/test
           cmake --build build/test -j9 --config Debug
           pushd build/test && ctest && popd

[chizkiyahu]

@kirklandsign Sorry for the delay. I updated the PR as suggested.

[kirklandsign]

Seems we don't have a corresponding docker image. Maybe use raw GH action instead of uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main wrapper