Skip to content

Commit

Permalink
yolo
Browse files Browse the repository at this point in the history
  • Loading branch information
msaroufim committed May 16, 2024
1 parent ba07163 commit 90f7c99
Showing 1 changed file with 25 additions and 21 deletions.
46 changes: 25 additions & 21 deletions .github/workflows/nightly-build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,24 +13,28 @@ on:

jobs:
build-and-publish:
- uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: "linux.g5.12xlarge.nvidia.gpu"
gpu-arch-type: "cuda"
gpu-arch-version: "12.1"
script: |
conda create -n venv python=3.9 -y
conda activate venv
echo "::group::Install newer objcopy that supports --set-section-alignment"
yum install -y devtoolset-10-binutils
export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH
python -m pip install --upgrade pip
pip install setuptools wheel twine
pip install torch
pip install -r requirements.txt
pip install -r dev-requirements.txt
export TORCHAO_NIGHTLY=1
python setup.py sdist bdist_wheel
pytest test --verbose -s
REPOSITORY_URL="https://upload.pypi.org/legacy/"
twine upload --repository-url $REPOSITORY_URL -u __token__ -p ${{ secrets.PYPI_API_TOKEN }} dist/
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2

- uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: "linux.g5.12xlarge.nvidia.gpu"
gpu-arch-type: "cuda"
gpu-arch-version: "12.1"
script: |
conda create -n venv python=3.9 -y
conda activate venv
echo "::group::Install newer objcopy that supports --set-section-alignment"
yum install -y devtoolset-10-binutils
export PATH=/opt/rh/devtoolset-10/root/usr/bin/:$PATH
python -m pip install --upgrade pip
pip install setuptools wheel twine
pip install torch
pip install -r requirements.txt
pip install -r dev-requirements.txt
export TORCHAO_NIGHTLY=1
python setup.py sdist bdist_wheel
pytest test --verbose -s
REPOSITORY_URL="https://upload.pypi.org/legacy/"
twine upload --repository-url $REPOSITORY_URL -u __token__ -p ${{ secrets.PYPI_API_TOKEN }} dist/

0 comments on commit 90f7c99

Please sign in to comment.