File tree Expand file tree Collapse file tree 3 files changed +82
-12
lines changed Expand file tree Collapse file tree 3 files changed +82
-12
lines changed Original file line number Diff line number Diff line change
1
+ #! /bin/bash
2
+
3
+ set -ex
4
+
5
+ install_buck () {
6
+ if ! command -v zstd & > /dev/null; then
7
+ brew install zstd
8
+ fi
9
+
10
+ curl https://github.com/facebook/buck2/releases/download/2023-07-18/buck2-x86_64-apple-darwin.zst -o buck2-x86_64-apple-darwin.zst
11
+ zstd -d buck2-x86_64-apple-darwin.zst -o buck2
12
+
13
+ chmod +x buck2
14
+ mv buck2 /opt/homebrew/bin
15
+
16
+ rm buck2-x86_64-apple-darwin.zst
17
+ }
18
+
19
+ install_conda_dependencies () {
20
+ pushd .ci/docker
21
+ # Install conda dependencies like flatbuffer
22
+ const install --file conda-env-ci.txt
23
+ popd
24
+ }
25
+
26
+ install_pip_dependencies () {
27
+ pushd .ci/docker
28
+ # Install all Python dependencies, including PyTorch
29
+ pip install --progress-bar off -r requirements-ci.txt
30
+
31
+ TORCH_VERSION=2.1.0.dev20230731
32
+ pip install --progress-bar off --pre torch==" ${TORCH_VERSION} " --index-url https://download.pytorch.org/whl/nightly/cpu
33
+ popd
34
+ }
35
+
36
+ install_buck
37
+ install_conda
38
+ install_pip_dependencies
Original file line number Diff line number Diff line change
1
+ #! /bin/bash
2
+
3
+ set -ex
4
+
5
+ install_executorch () {
6
+ # Install executorch, this assumes that Executorch is checked out in the
7
+ # current directory
8
+ pip3 install .
9
+ # Just print out the list of packages for debugging
10
+ pip3 list
11
+ }
12
+
13
+ build_and_test_executorch () {
14
+ # Build executorch runtime
15
+ buck2 build //examples/executor_runner:executor_runner
16
+ # Export a test model
17
+ python3 -m examples.export.export_example --model_name=" linear"
18
+ # Run test model
19
+ buck2 run //examples/executor_runner:executor_runner -- --model_path ./linear.ff
20
+ }
21
+
22
+ install_executorch
23
+ build_and_test_executorch
Original file line number Diff line number Diff line change @@ -12,8 +12,8 @@ concurrency:
12
12
cancel-in-progress : true
13
13
14
14
jobs :
15
- buck-build-test :
16
- name : buck-build-test
15
+ buck-build-test-linux :
16
+ name : buck-build-test-linux
17
17
uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
18
18
with :
19
19
runner : linux.2xlarge
@@ -27,14 +27,23 @@ jobs:
27
27
# here, as it's there in the container
28
28
export PATH="/opt/conda/envs/py_${PYTHON_VERSION}/bin:${PATH}"
29
29
30
- # Install executorch
31
- pip3 install .
32
- # Just print out the list of packages for debugging
33
- pip3 list
30
+ # Build and test Executorch
31
+ bash .ci/scripts/test.sh
34
32
35
- # Build executorch runtime
36
- buck2 build //examples/executor_runner:executor_runner
37
- # Export a test model
38
- python3 -m examples.export.export_example --model_name="linear"
39
- # Run test model
40
- buck2 run //examples/executor_runner:executor_runner -- --model_path ./linear.ff
33
+ buck-build-test-macos :
34
+ name : buck-build-test-macos
35
+ uses : pytorch/test-infra/.github/workflows/macos_job.yml@main
36
+ with :
37
+ runner : macos-m1-12
38
+ submodules : ' true'
39
+ ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
40
+ script : |
41
+ WORKSPACE=$(pwd)
42
+
43
+ pushd "${WORKSPACE}/pytorch/executorch"
44
+ # Setup MacOS dependencies as there is no Docker support on MacOS atm
45
+ bash .ci/scripts/setup-macos.sh
46
+
47
+ # Build and test Executorch
48
+ bash .ci/scripts/test.sh
49
+ popd
You can’t perform that action at this time.
0 commit comments