Add OSS CI workflow for MacOS M1 (#32)

Summary:
Pull Request resolved: #32

This builds and runs Executorch on MacOS M1, similar to the Linux x86_64 job

Differential Revision: D47929778

fbshipit-source-id: bb7916378aad85cc60c69d6372e00c4bb6104fab

Author: huydhn

.ci/docker/requirements-ci.txt

@@ -1,4 +1,6 @@
 mpmath==1.3.0
+numpy==1.25.2
 PyYAML==6.0.1
 ruamel.yaml==0.17.32
 sympy==1.12
+zstd==1.5.5.1

.ci/scripts/setup-macos.sh

@@ -0,0 +1,49 @@
+#!/bin/bash
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+set -ex
+
+install_buck() {
+  if ! command -v zstd &> /dev/null; then
+    brew install zstd
+  fi
+
+  if ! command -v wget &> /dev/null; then
+    brew install wget
+  fi
+
+  if ! command -v buck2 &> /dev/null; then
+    wget -q https://github.com/facebook/buck2/releases/download/2023-07-18/buck2-x86_64-apple-darwin.zst
+    zstd -d buck2-x86_64-apple-darwin.zst -o buck2
+
+    chmod +x buck2
+    mv buck2 /opt/homebrew/bin
+
+    rm buck2-x86_64-apple-darwin.zst
+  fi
+}
+
+install_conda() {
+  pushd .ci/docker
+  # Install conda dependencies like flatbuffer
+  const install --file conda-env-ci.txt
+  popd
+}
+
+install_pip_dependencies() {
+  pushd .ci/docker
+  # Install all Python dependencies, including PyTorch
+  pip install --progress-bar off -r requirements-ci.txt
+
+  TORCH_VERSION=2.1.0.dev20230731
+  pip install --progress-bar off --pre torch=="${TORCH_VERSION}" --index-url https://download.pytorch.org/whl/nightly/cpu
+  popd
+}
+
+install_buck
+install_conda
+install_pip_dependencies

.ci/scripts/test.sh

@@ -0,0 +1,28 @@
+#!/bin/bash
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+set -ex
+
+install_executorch() {
+  # Install executorch, this assumes that Executorch is checked out in the
+  # current directory
+  pip3 install .
+  # Just print out the list of packages for debugging
+  pip3 list
+}
+
+build_and_test_executorch() {
+  # Build executorch runtime
+  buck2 build //examples/executor_runner:executor_runner
+  # Export a test model
+  python3 -m examples.export.export_example --model_name="linear"
+  # Run test model
+  buck2 run //examples/executor_runner:executor_runner -- --model_path ./linear.ff
+}
+
+install_executorch
+build_and_test_executorch

.github/workflows/pull.yml

@@ -12,8 +12,8 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  buck-build-test:
-    name: buck-build-test
+  buck-build-test-linux:
+    name: buck-build-test-linux
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: linux.2xlarge
@@ -27,14 +27,23 @@ jobs:
         # here, as it's there in the container
         export PATH="/opt/conda/envs/py_${PYTHON_VERSION}/bin:${PATH}"
 
-        # Install executorch
-        pip3 install .
-        # Just print out the list of packages for debugging
-        pip3 list
+        # Build and test Executorch
+        bash .ci/scripts/test.sh
 
-        # Build executorch runtime
-        buck2 build //examples/executor_runner:executor_runner
-        # Export a test model
-        python3 -m examples.export.export_example --model_name="linear"
-        # Run test model
-        buck2 run //examples/executor_runner:executor_runner -- --model_path ./linear.ff
+  buck-build-test-macos:
+    name: buck-build-test-macos
+    uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
+    with:
+      runner: macos-m1-12
+      submodules: 'true'
+      ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
+      script: |
+        WORKSPACE=$(pwd)
+
+        pushd "${WORKSPACE}/pytorch/executorch"
+        # Setup MacOS dependencies as there is no Docker support on MacOS atm
+        bash .ci/scripts/setup-macos.sh
+
+        # Build and test Executorch
+        bash .ci/scripts/test.sh
+        popd