-
Notifications
You must be signed in to change notification settings - Fork 175
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add new workflow to build commits and output to s3
- slightly different than existing one
- Loading branch information
Raunak Bhagat
committed
Nov 18, 2024
1 parent
274f300
commit f0c99eb
Showing
2 changed files
with
96 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,31 @@ | ||
name: Install uv, rust, and python | ||
description: Install uv, rust, and python | ||
runs: | ||
using: composite | ||
steps: | ||
- name: Install rust | ||
shell: bash | ||
run: | | ||
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y | ||
CARGO_BIN="$HOME/.cargo/bin" | ||
echo 'export PATH="$CARGO_BIN:$PATH"' >> $HOME/.bashrc | ||
echo "$CARGO_BIN" >> $GITHUB_PATH | ||
- name: Install uv | ||
shell: bash | ||
run: | | ||
curl -LsSf https://astral.sh/uv/install.sh | sh | ||
UV_BIN="$HOME/.local/bin" | ||
echo 'export PATH="$UV_BIN:$PATH"' >> $HOME/.bashrc | ||
echo "$UV_BIN" >> $GITHUB_PATH | ||
- name: Source .bashrc | ||
shell: bash | ||
run: | | ||
source $HOME/.bashrc | ||
- name: Install python (version 3.9) | ||
shell: bash | ||
run: | | ||
uv python install 3.9 | ||
uv python pin 3.9 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,65 @@ | ||
name: Build a Daft commit and store the outputted wheel in AWS S3 | ||
|
||
on: | ||
workflow_dispatch: | ||
workflow_call: | ||
secrets: | ||
ACTIONS_AWS_ROLE_ARN: | ||
description: The ARN of the AWS role to assume | ||
required: true | ||
inputs: | ||
commit: | ||
type: string | ||
description: The commit hash to build | ||
required: true | ||
outputs: | ||
wheel: | ||
description: The wheel file that was built | ||
value: ${{ jobs.build-commit.outputs.wheel }} | ||
|
||
jobs: | ||
build-commit: | ||
runs-on: buildjet-8vcpu-ubuntu-2004 | ||
timeout-minutes: 15 # Remove for ssh debugging | ||
permissions: | ||
id-token: write | ||
contents: read | ||
outputs: | ||
wheel: ${{ steps.build_and_upload.outputs.wheel }} | ||
steps: | ||
- uses: aws-actions/configure-aws-credentials@v4 | ||
with: | ||
aws-region: us-west-2 | ||
role-to-assume: ${{ secrets.ACTIONS_AWS_ROLE_ARN }} | ||
role-session-name: daft-performance-comparisons-build | ||
- uses: actions/checkout@v4 | ||
with: | ||
ref: ${{ inputs.commit || github.sha }} | ||
fetch-depth: 1 | ||
- uses: ./.github/actions/install | ||
- uses: buildjet/cache@v4 | ||
with: | ||
path: ~/target | ||
key: ${{ runner.os }}-cargo-deps-${{ hashFiles('**/Cargo.lock') }} | ||
restore-keys: ${{ runner.os }}-cargo-deps- | ||
- name: Build the python wheel and upload it to AWS S3 | ||
id: build_and_upload | ||
run: | | ||
if ! ls ~/target/wheels/*.whl 1> /dev/null 2>&1; then | ||
# Build wheel | ||
export CARGO_TARGET_DIR=~/target | ||
uv v | ||
source .venv/bin/activate | ||
uv pip install pip maturin | ||
maturin build --release | ||
fi | ||
# Upload wheel | ||
# (there should only be one output wheel in this directory) | ||
for file in ~/target/wheels/*.whl; do | ||
aws s3 cp $file s3://github-actions-artifacts-bucket/builds/${{ inputs.commit || github.sha }}/ --acl public-read --no-progress; | ||
file_basename=$(basename $file) | ||
echo "wheel=$file_basename" >> $GITHUB_OUTPUT | ||
echo "Output wheel has been built and stored in S3 at the following location:" >> $GITHUB_STEP_SUMMARY | ||
echo "https://us-west-2.console.aws.amazon.com/s3/buckets/github-actions-artifacts-bucket?prefix=builds/${{ inputs.commit || github.sha }}/" >> $GITHUB_STEP_SUMMARY | ||
done |